ModuleNotFoundError No module named ‘torchvision.models.utils‘

在DBNet.pytorch训练脚本中遇到了' torchvision.models.utils'模块未找到的问题。通过将导入语句从`from torchvision.models.utils import load_state_dict_from_url`更改为`from torch.hub import load_state_dict_from_url`解决了此问题。
摘要由CSDN通过智能技术生成
  File "/home/ps/wanghao/DBNet.pytorch-master/tools/train.py", line 78, in <module>
    main(config)
  File "/home/ps/wanghao/DBNet.pytorch-master/tools/train.py", line 24, in main
    from models import build_model, build_loss
  File "/home/ps/wanghao/DBNet.pytorch-master/models/__init__.py", line 5, in <module>
    from .model import Model
  File "/home/ps/wanghao/DBNet.pytorch-master/models/model.py", line 8, in <module>
    from models.backbone import build_backbone
  File "/home/ps/wanghao/DBNet.pytorch-master/models/backbone/__init__.py", line 7, in <module>
    from .shufflenetv2 import *
  File "/home/ps/wanghao/DBNet.pytorch-master/models/backbone/shufflenetv2.py", line 7, in <module>
    from torchvision.models.utils import load_state_dict_from_url
ModuleNotFoundError: No module named 'torchvision.models.utils'

from torchvision.models.utils import load_state_dict_from_url

改为:

from torch.hub import load_state_dict_from_url
  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值