Torchvision 0.6.0 doesn't work on Ubuntu 16.04.3 with Python 3.5.2
https://github.com/pytorch/vision/issues/2132arrow-up-right
module.to()
move to a specific device
print(module) 可以把一个模型的组成按照树的结构打印出来。
Last updated 4 years ago