无法从“ResNet50_Weights”导入名称“torchvision.models.resnet”

Question

以前，我成功地加载了一个带有ResNet参数的ResNet50_Weights模型，但是突然间，我开始得到以下错误：

Traceback (most recent call last):
  File "splitting_models.py", line 3, in <module>
    from torchvision.models.resnet import ResNet50_Weights
ImportError: cannot import name 'ResNet50_Weights' from 'torchvision.models.resnet' (/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/torchvision/models/resnet.py)

以下是进口：

from torchvision.models import ResNet50_Weights

我怎么才能解决这个问题？

PyTorch版本: 1.2.0

TorchVision版本: 0.4.0

编辑

升级使用

pip install --upgrade torch torchvision

以下版本修复了此问题：

PyTorch版本: 1.12.0

TorchVision版本: 0.13.0

Answer 1

这个问题花了我几天时间才解决。在运行包含Py火炬模型的代码之前，请确保您已连接到稳定的网络。这是因为当您第一次运行像resnet50、alexnet、resnet18这样的Pytorch模型时，它下载它的功能，所以在安装错误的情况下，如果您试图重新运行，它会缓存它的下载并绘制这样的错误。要解决这个问题，请删除缓存文件所在的位置，并尝试重新运行具有稳定网络的代码。就我的情况而言：

C:\Users\user/.cache\torch\hub\checkpoints\resnet18-f37072fd.pth

这是is缓存文件所在的地方，删除它并尝试使用稳定的网络重新运行。

我希望这会有所帮助。谢谢你，

Answer 2

您可以尝试将torchvision升级/降级到不同的版本。它可能会解决这个问题。