pytorch加载模型时出现.....ckpt_100.pth is a zip archive (did you mean to use torch.jit.load()?)
在测试加载训练好的模型时出现上方问题,参考这篇文章,原因是训练和测试的torch版本不一致。
训练的时候是1.6,测试的时候是1.2,因此需要先在1.6版本下加载模型,重新保存,在保存的时候设置use_new_zipfile_serialization=False 就行了。
原本代码:
checkpoint = torch.load(val_path+args.resume)
net.load_state_dict(checkpoint['state_dict'])
重新保存的代码:
checkpoint = torch.load(val_path+args.resume)
state_dict = net.state_dict()
torch.save({'state_dict':state_dict},val_path+args.resume,_use_new_zipfile_serialization=False)
pytorch加载模型时出现.....ckpt_100.pth is a zip archive (did you mean to use torch.jit.load()?)相关推荐
- pytorch版本问题RuntimeError: params/unet.pth is a zip archive (did you mean to use torch.jit.load()?)
pytorch版本问题RuntimeError: params/unet.pth is a zip archive (did you mean to use torch.jit.load()?) 问题 ...
- RuntimeError: xxx.pth is a zip archive (did you mean to use torch.jit.load()?)
用高版本的pytorch训练模型,在低版本中测试时出现如下错误: RuntimeError: all_ep_296.pth is a zip archive (did you mean to use ...
- .pth is a zip archive (did you mean to use torch.jit.load()?)
RuntimeError: bert.pth is a zip archive (did you mean to use torch.jit.load()?) .pth is a zip archiv ...
- xxx.pth或者xxx.pt is a zip archive(did you mean to use torch.jit.load()?问题解决
提示:推荐使用方法三,不会"科学上网"的请关注微信公众号"DGX杂学"并在后台私信! 错误:使用torch.load()加载模型参数时,提示"xxx. ...
- Pytorch加载模型只导入部分层权重,即跳过指定网络层的方法
需求 Pytorch加载模型时,只导入部分层权重,跳过部分指定网络层.(权重文件存储为dict形式) 方法一 常见方法:加载权重时用if对网络层进行筛选 ''' # model为定义的网络结构: cl ...
- PyTorch加载模型model.load_state_dict()问题,Unexpected key(s) in state_dict: “module.features..,Expected .
希望将训练好的模型加载到新的网络上.如上面题目所描述的,PyTorch在加载之前保存的模型参数的时候,遇到了问题. Unexpected key(s) in state_dict: "mod ...
- Pytorch加载模型并进行图像分类预测
目录 1. 整体流程 1)实例化模型 2)加载模型 3)输入图像 4)输出分类结果 5)完整代码 2. 处理图像 1) How can i convert an RGB image into gray ...
- Gazebo加载模型时黑屏
Gazebo加载模型时黑屏 1. 黑屏状态 2. 解决办法1 3. 解决办法2 1. 黑屏状态 Gazebo加载模型的时候会发现一直处于这种状态 这可能是因为model库加载不正确导致的 2. 解决办 ...
- pytorch加载模型报错Unexpected key(s) in state_dict: module.conv1.weight, module.bn1
文章目录 背景 报错 原因 解决 背景 Pytorch在加载模型参数的时候,有两种情况可能出现这种问题: 自己写的网络结构,例如: 代码 import models arch = 'resnet50' ...
最新文章
- ARM架構的Linux內核的空間佈局
- Java面试常被问到的题目+解答
- 为你的博客添加几分色彩
- java 浮点数精度_Java中浮点数精度问题
- oracle constraint_type 问题
- fir.im Log Guru 正式开源,快速找到 iOS 应用无法安装的原因
- 1月全球Web服务器市场:Apache份额回升至41.64%
- C++虚函数表的总结
- mongodb创建集合命令db.createCollection详解
- IIS 添加mime 支持 apk,exe,.woff,IIS MIME设置 ,Android apk下载的MIME 设置 苹果ISO .ipa下载mime 设置...
- totolink服务器未响应,TOTOLINK路由器设置后无法上网问题的解决方法
- 专题六:数值微积分与方程求解
- 卖零食怎么引流?怎么推广自己的零食产品,零食店怎么做引流?
- 前大疆RoboMaster技术总监:机器人工程师学习计划
- visual assist x太卡了_河西区专业达格XD7500租赁 X-RAY -哪家好-老友网
- Microsoft visual studio关闭安全检查的几种方法
- 20-统一网关Gateway-全局过滤器
- 前端---HTML制作百度首页
- pyPDF2或pyPDF4做pdf页面提取出错
- JAVA 身份证校验与统一社会信用代码校验
热门文章
- 电脑全屏时任务栏没有隐藏,解决办法
- DataGridView背景图片
- 基于消失点的相机自标定(1)
- 【平头哥蓝牙Mesh网关开发套件试用体验】PHY6220 蓝牙键盘
- python web自动化demo阿里巴巴自动向商家发送消息
- Uipath如何获取outlook邮件附件/特定类型附件
- 计算机技术调查过程,计算机技术在质量管理中的一项应用——用计算机绘制直方图和进行工序能力调查...
- Python实践 - 网络爬虫笔记 - 2、从网站上爬取公开信息
- 【历史上的今天】1 月 9 日:iPhone 15 周年;iTunes 问世;激光打印机的发明者出生
- 五阿哥钢铁平台Docker容器云平台建设经验分享——你想知道的这里都有!!