美文网首页
快速在服务器中部署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