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