Pytorch_使torchvision只从本地加载代码,不再自动下载模型
最近在windows系统下使用pytorch1.5.0和torchvision0.6.1做一个小项目(使用torchvision内置的fasterrcnn模型进行目标检测)的时候发现一个小问题,即torchversion会自动下载一个resnet50的模型到一个系统路径:C:\Users\**你的windows用户名**\.cache\torch\checkpoints下,这种内置方法既不方便(模型的下载源在国外,下载比较慢),也不好部署,网上也查到一些方法,包括:pytorch预训练模型的下载地址以及解决
原创
2020-09-15 09:47:48 ·
1250 阅读 ·
0 评论