最困难 的一步对我来说是自动下载预训练模型。
(base) wmz@wmz:~$ cd /home/wmz/.cache/torch/checkpoints
(base) wmz@wmz:~/.cache/torch/checkpoints$ ls
tmpnfltw32z
(base) wmz@wmz:~/.cache/torch/checkpoints$ cp /home/wmz/Downloads/resnet152-b121ed2d.pth ./
默认下载到下面的路径下
/home/wmz/.cache/torch/checkpoints
在执行train.py时,会显示下载模型路径,实际网速很慢很难下载成功,我的操作是先下载好,然后存放到上面的路径下。
然后就可以听到机器轰鸣,训练跑起来了,不一会儿机器就发烫了。