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

Question

我之前成功加载了带有 ResNet50_Weights 参数的 ResNet 模型，但突然开始出现以下错误：

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

火炬视觉版本：0.4.0

编辑

升级使用

pip install --upgrade torch torchvision

以下版本修复了该问题：

PyTorch 版本：1.12.0

火炬视觉版本：0.13.0

Answer 1

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