ModuleNotFoundError No module named ‘torchvision.models.utils‘

  File "/home/ps/wanghao/DBNet.pytorch-master/tools/train.py", line 78, in 
    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 
    from .model import Model
  File "/home/ps/wanghao/DBNet.pytorch-master/models/model.py", line 8, in 
    from models.backbone import build_backbone
  File "/home/ps/wanghao/DBNet.pytorch-master/models/backbone/__init__.py", line 7, in 
    from .shufflenetv2 import *
  File "/home/ps/wanghao/DBNet.pytorch-master/models/backbone/shufflenetv2.py", line 7, in 
    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

你可能感兴趣的:(java,java,后端)