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

最新推荐文章于 2024-10-15 10:57:24 发布

原创最新推荐文章于 2024-10-15 10:57:24 发布 · 4.2k 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#python #pytorch

当尝试在不同项目间迁移PyTorch模型时，可能会遇到`ModuleNotFoundError`，这通常是因为缺少保存模型时的模块引用。解决方案是在加载模型的项目中引入保存模型项目的路径。此外，将模型转换为 TorchScript 虽然能解决加载问题，但可能引发新的错误。正确做法是在加载端导入保存权重的项目路径。

部署运行你感兴趣的模型镜像

如果保存模型时使用

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)

您可能感兴趣的与本文相关的镜像