Miscellaneous

  • Torchvision 0.6.0 doesn't work on Ubuntu 16.04.3 with Python 3.5.2

  • module.to()

    • move to a specific device

  • print(module) 可以把一个模型的组成按照树的结构打印出来。

Last updated

Was this helpful?