1、新版本torchvision将load_state_dict_from_url移到了torch中
将
from torchvision.models.utils import load_state_dict_from_url
改成
from torch.hub import load_state_dict_from_url
2、或者安装低版本torchvision
安装相应的 torchvision 版本,目前实测 0.4.0~0.7.0 的版本均支持此类语句。
pip install torchvision==0.4
参考:
(17条消息) No module named ‘torchvision.models.utils‘_yyywxk的博客-CSDN博客