没有名为“torchvision.models.utils”的模块

Question

当我使用pytorch=1.10.0、torchvision=0.11.1的环境运行代码时，运行到了from torchvision.models.utils import load_state_dict_from_url 的语句。出现以下错误时会出现：

>>> from torchvision.models.utils import load_state_dict_from_url
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'torchvision.models.utils'

Answer 1

查阅torchvision的代码库后，有一个解决方案：

请注意，此语法仅适用于更高版本的 PyTorch。

.utils import load_state_dict_from_url 的原始代码不适用。您无法从 .utils 导入 load_state_dict_from_url。

将 .utils 更改为 torch.hub 可以解决该问题。

from torch.hub import load_state_dict_from_url

这对我有用。