解决cannot import name 'load_state_dict_from_url'

介绍

  最近使用pytorch搭建YOLOv4模型时发现load_state_dict_from_url无法导入，导入的代码是：
    from torchvision.models.utils import load_state_dict_from_url
  提示出错为：
    cannot import name 'load_state_dict_from_url'

修改

  将from torchvision.models.utils import load_state_dict_from_url注释掉，改成以下代码：
    

try:
    from torch.hub import load_state_dict_from_url
except ImportError:
    from torch.utils.model_zoo import load_url as load_state_dict_from_url

  最后成功运行