ModuleNotFoundError: No module named ‘torchvision.models.utils‘

Posted 2022-03-22 AI浩

  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