小编Why*_*rch的帖子

我正在尝试使用 torch.load 加载预训练模型。

我收到以下错误：

ModuleNotFoundError: No module named 'utils'

我通过从命令行打开它来检查我使用的路径是否正确。可能是什么原因造成的？

这是我的代码：

import torch
import sys

PATH = './gan.pth'

model = torch.load(PATH)
model.eval()

编辑：整个错误堆栈：

Traceback (most recent call last):
  File "load.py", line 6, in <module>
    model = torch.load(PATH)
  File "C:\Users\user\anaconda3\envs\pytorch-flask\lib\site-packages\torch\serialization.py", line 595, in load
    return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
  File "C:\Users\user\anaconda3\envs\pytorch-flask\lib\site-packages\torch\serialization.py", line 774, in _legacy_load
    result = unpickler.load()
ModuleNotFoundError: No module named 'utils'