快速在服务器中部署Torchversion代码训练模型
2020-10-13 本文已影响0人
CPinging
1 在系统环境中安装torchverison（当然之前的pytorch应该装好，之后pip）
2 进入下面链接并下载到自己服务器上
https://github.com/Pinging-ZJU/examples
3 之后进入ImageNet文件夹。运行main函数：
python3 main.py -a alexnet ~/DataSet/tiny-imagenet-200 --lr 0.01
其中~/DataSet/tiny-imagenet-200为ImageNet的dir。之后就看到模型跑起来了
4 如果想改model的内容，可以用pip3 show torchvision，之后找到对应的路径
image.png
根据该路径进入到xxxxxxx/torchvision/models里面便可以修改。
