关于result = unpickler.load()ModuleNotFoundError: No module named ‘models‘解决

如果保存模型时使用

torch.save(model, path)

在加载时使用

model = torch.load(path)

可能会出现如下错误，通常这种错误就是把一个项目保存的权重文件用在另一个项目中进行加载，

result = unpickler.load()
ModuleNotFoundError: No module named 'models'

我尝试用官方示例代码把模型转为torch script来解决

scripted_module = torch.jit.script(MyModule())
torch.jit.save(scripted_module, 'mymodule.pt')
torch.jit.load('mymodule.pt')

然后加载没有问题，但是又出现后续其他错误，没有使用该方法。

解决办法：

在加载权重的项目代码中引入保存权重的项目代码的总路径。

import sys
sys.path.insert(0, '/home/hpc/PycharmProjects/Siam-NestedUNet/')


"""
/home/hpc/PycharmProjects/Siam-NestedUNet/是一个项目路径，
这个项目保存了权重checkpoint_epoch_59.pt,我把这个权重复制到另一个项目中加载
需要加上注释上面2行的代码
"""
model_path = "checkpoint_epoch_59.pt"
model = torch.load(model_path)