关于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)

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

Pytorch ModuleNotFoundError: No module named ‘models‘
问题今天,在使用pytorch深度学习框架训练出来的模型文件,在另外的工程中使用,碰到如下的错误: (shennong) ubuntu@ubuntu:~/zy/medicinebox_detecto ...
jupyter notebook运行出错：ModuleNotFoundError: No module named ‘keras‘ 解决办法
接上一篇 <win10+python3.6+tensorflow-cpu+keras+Pycharm环境下的tensorflow配置方法> 通过Anaconda Prompt 打开查看当 ...
[Dynamic Language] Python3.7 源码安装 ModuleNotFoundError: No module named '_ctypes' 解决记录...
Python3.7 源码安装 ModuleNotFoundError: No module named '_ctypes' 解决记录源码安装时报错 File "/home/abeenser ...
python程序在命令行执行提示ModuleNotFoundError: No module named ‘XXX‘ 解决方法
python程序在命令行执行提示ModuleNotFoundError: No module named 'XXX' 解决方法参考文章: (1)python程序在命令行执行提示ModuleNotFo ...
Python安装常见问题：ModuleNotFoundError: No module named ‘_ctypes‘ 解决办法
Python安装常见问题:ModuleNotFoundError: No module named '_ctypes' 解决办法参考文章: (1)Python安装常见问题:ModuleNotFoun ...
ModuleNotFoundError: No module named ‘matplotlib‘ 解决办法
问题:ModuleNotFoundError: No module named 'matplotlib' 解决办法: round 1:打开PyCharm,点击File->Settings,点击弹 ...
导入自己模块包导入报错：ModuleNotFoundError: No module named ‘xxx‘解决方法晓物智联
导入自己模块包导入报错:ModuleNotFoundError: No module named 'xxx'解决方法文章来源于:http://www.52phm.cn/blog/detail/28 ...
pip报错ModuleNotFoundError: No module named ‘dataclasses‘解决
pip报错ModuleNotFoundError: No module named 'dataclasses'解决问题描述使用pip无论输入何种指令都会出现如下报错,见下截图初步的解决思路是重装 ...
Python遇错：ModuleNotFoundError: No module named ‘IPython‘解决办法。
遇到情况: ModuleNotFoundError: No module named 'IPython' 解决办法: 1.打开pycharm终端: 2.依次输入 pip3 install Ipytho ...

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

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

最新文章

热门文章