问题描述

今天我研究 Siamese-pytorch 孪生神经网络的时候，根据教程运行 predict.py，就报错如下：

Traceback (most recent call last):
  File "F:/验证码项目/Siamese-pytorch-master/predict.py", line 4, in <module>
    from siamese import Siamese
  File "F:\验证码项目\Siamese-pytorch-master\siamese.py", line 7, in <module>
    from nets.siamese import Siamese as siamese
  File "F:\验证码项目\Siamese-pytorch-master\nets\siamese.py", line 4, in <module>
    from nets.vgg import VGG16
  File "F:\验证码项目\Siamese-pytorch-master\nets\vgg.py", line 3, in <module>
    from torchvision.models.utils import load_state_dict_from_url
ModuleNotFoundError: No module named 'torchvision.models.utils'

问题原因

根据项目说明文档，需要 torch==1.2.0。但是我的环境比较新，torch==1.13.1+cu116，torchvision==0.14.1+cu116。所以是因为torchvision版本过高导致的错误。

解决方法

将from torchvision.models.utils import load_state_dict_from_url
改为：from torch.hub import load_state_dict_from_url