美文网首页
快速在服务器中部署Torchversion代码训练模型

快速在服务器中部署Torchversion代码训练模型

作者: CPinging | 来源:发表于2020-10-13 12:28 被阅读0次

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里面便可以修改。

相关文章

网友评论

      本文标题:快速在服务器中部署Torchversion代码训练模型

      本文链接:https://www.haomeiwen.com/subject/hnkvpktx.html