pytorch 1.7.x训练保存的模型在1.4低版本无法加载

pytorch1.6.x及更高版本，在保存模型和低版本有区别，因此导致高版本的保存模型在低版本下加载失败的问题:

我们只需在较高版本torch下保存时，设置_use_new_zipfile_serialization=False

weights = 'xxx.pth'
temp = torch.load(weights)
...
data = ... #type:OrderedDict
torch.save(data, 'new_model.pth', _use_new_zipfile_serialization=False)

关于data的获取不同方式保存的模型处理方式不太一致，要搞清自己的模型是怎么保存的，常见的几种保存形式可以参考我的这篇博客：
https://blog.csdn.net/qq_41368074/article/details/120812819
一个具体例子可以参考：
https://blog.csdn.net/flying_ant2018/article/details/115403774

pytorch 1.7.x训练保存的模型在1.4低版本无法加载相关推荐

pytorch 1.7训练保存的模型在1.4低版本无法加载：frame #63: ＜unknown function＞ + 0x1db3e0 (0x55ba98ddd3e0 in /data/user
pytorch 1.7高版本训练保存的模型在1.4低版本无法加载,报错: torch.load('/home/user1/model_best_b.pth.tar') Traceback (most ...
单卡加载多卡训练保存的模型
1.问题直接用加载单卡模型的代码来加载多卡训练保存的模型时会报这样一个错误: RuntimeError: Error(s) in loading state_dict for : Missing k ...
keras训练完以后怎么预测_一文告诉你如何将Keras模型保存到文件中，并再次加载它们来进行预测。...
Keras是一个用于深度学习的简单而强大的Python库. 鉴于深度学习模式可能需要数小时.数天甚至数周的时间来培训,了解如何保存并将其从磁盘中加载是很重要的. 在本文中,您将发现如何将Keras模型 ...
“Word自动更改后的内容保存到通用文档模板上。是否加载该模板？“的解决办法...
在win7系统下,Word2010出现了不能正常关闭.打开一个已有word文档,点击右上角关闭按钮后,先提示"word已停止工作,windows正在检查该问题的解决方案",随后提示 ...
初次转化max模型为3D Tiles失败、cesium加载一个obj格式的3D 机房模型
先下一个3D max模型,后缀为.max:用3D max打开:网上下的其他模型查看器不能打开: 打开时会出错误提示:对max模型文件和3d max还不了解: 打开后如下:是一个机房的3D模型: 下载一 ...
three相机在模型上_实例讲解Three.js加载外部模型
本文主要介绍了Three.js外部模型加载的教程详解,在文章给大家补充介绍了three.js 外部模型加载json的方法,非常不错,具有参考借鉴价值,需要的朋友可以参考下,希望能帮助到大家. 1. 首 ...
babylonjs 分部加载模型_17 Babylonjs入门进阶使用场景加载器加载glTF，OBJ，STL模型...
Babylon.js内置的模型格式是.babylon,Babylon.js可以不需要其它额外的插件即可加载. 注意:由于你导入的模型可以具有rotationQuaternion的设置,因此再设置rot ...
cesium获取模型高度_Cesium中地形数据的加载
Cesium开发中,如果想要看到真实感,地形数据(DEM)不可或缺.但是很多非GIS专业的人,对地形数据的定位不清晰,不明白地形数据如何展示. 最近很多人问我这个问题,综合看下来,主要问题就集中在地形 ...
python加载模型包占用内存多大_加载pickle python对象会占用大量内存
我有一个 python的pickle对象,它生成一个180 Mb的文件.当我取消它时,内存使用量会爆炸到2或3Gb.你有类似的经历吗?这是正常的吗? 对象是包含字典的树:每个边是一个字母,每个节点都是 ...

pytorch 1.7.x训练保存的模型在1.4低版本无法加载

pytorch 1.7.x训练保存的模型在1.4低版本无法加载相关推荐

最新文章

热门文章