解决ModuleNotFoundError: No module named ‘models‘

当使用训练好的模型进行:

torch.load('./x.pt')

发生以上报错。

原因可能是该模型仅保存了参数而非整体模型结构(可能原因,不确定)

因此需要将原始定义模型的py文件移植到同一目录下即可解决

你可能感兴趣的:(pytorch,python,人工智能)