学习使用torch hub,但是遇到一些问题如: ImportError: cannot import name 'resnext50_32x4d'
之前知道torch hub这个库比较好用,而且看起来使用非常简单,但是自己一操作就报了标题错误;
torch版本我使用1.3.1, torchvison 却是0.2.1
顺着报错的提示找到torchvision.models 导入模型的时候缺少某些模型
于是选择更新torchvision 到最新版0.4.2
问题解决!
感觉torch hub 和torchvison.models 中导入...
原创
2020-03-04 10:54:12 ·
6014 阅读 ·
1 评论