解决pytorch 保存模型遇到的问题


Posted in Python onMarch 03, 2021

今天用pytorch保存模型时遇到bug

Can't pickle <class 'torch._C._VariableFunctions'>

在google上查找原因,发现是保存时保存了整个模型的原因,而模型中有一些自定义的参数

torch.save(model,save_path) 改为 torch.save(model.state_dict(),save_path)

然后载入模型也做相应的更改就好了

补充:pytorch训练模型的一些坑

1. 图像读取

opencv的python和c++读取的图像结果不一致,是因为python和c++采用的opencv版本不一样,从而使用的解码库不同,导致读取的结果不同。

2. 图像变换

PIL和pytorch的图像resize操作,与opencv的resize结果不一样,这样会导致训练采用PIL,预测时采用opencv,结果差别很大,尤其是在检测和分割任务中比较明显。

3. 数值计算

pytorch的torch.exp与c++的exp计算,10e-6的数值时候会有10e-3的误差,对于高精度计算需要特别注意,比如

两个输入5.601597, 5.601601, 经过exp计算后变成270.85862343143174, 270.85970686809225

以上为个人经验,希望能给大家一个参考,也希望大家多多支持三水点靠木。如有错误或未考虑完全的地方,望不吝赐教。

Python 相关文章推荐
python获取标准北京时间的方法
Mar 24 Python
一篇文章入门Python生态系统(Python新手入门指导)
Dec 11 Python
python变量不能以数字打头详解
Jul 06 Python
Python实现对字符串的加密解密方法示例
Apr 29 Python
Python学习笔记之if语句的使用示例
Oct 23 Python
基于Django的ModelForm组件(详解)
Dec 07 Python
Python+OpenCV图片局部区域像素值处理改进版详解
Jan 23 Python
python导入坐标点的具体操作
May 10 Python
Python中函数的返回值示例浅析
Aug 28 Python
解决Python pip 自动更新升级失败的问题
Feb 21 Python
matplotlib 三维图表绘制方法简介
Sep 20 Python
PYTHON基于Pyecharts绘制常见的直角坐标系图表
Apr 28 Python
解决pytorch 模型复制的一些问题
Mar 03 #Python
Pytorch模型迁移和迁移学习,导入部分模型参数的操作
Mar 03 #Python
pytorch 实现L2和L1正则化regularization的操作
Mar 03 #Python
Pytorch自定义Dataset和DataLoader去除不存在和空数据的操作
Mar 03 #Python
python爬取youtube视频的示例代码
Mar 03 #Python
pytorch Dataset,DataLoader产生自定义的训练数据案例
Mar 03 #Python
解决pytorch 数据类型报错的问题
Mar 03 #Python
You might like
php无限极分类递归排序实现方法
2014/11/11 PHP
PHP实现将科学计数法转换为原始数字字符串的方法
2014/12/16 PHP
简单了解WordPress开发中update_option()函数的用法
2016/01/11 PHP
PHP设计模式之原型设计模式原理与用法分析
2018/04/25 PHP
javascript读取xml
2006/11/04 Javascript
Jquery实现Div上下移动示例
2014/04/23 Javascript
js和jquery设置disabled属性为true使按钮失效
2014/08/07 Javascript
JS应用正则表达式转换大小写示例
2014/09/18 Javascript
jquery结合CSS使用validate实现漂亮的验证
2015/01/29 Javascript
jQuery获得包含margin的outerWidth和outerHeight的方法
2015/03/25 Javascript
Jquery和BigFileUpload实现大文件上传及进度条显示
2016/06/27 Javascript
微信小程序 获取当前地理位置和经纬度实例代码
2016/12/05 Javascript
jquery实现下拉框左右选择功能
2017/02/21 Javascript
vue-cli开发时,关于ajax跨域的解决方法(推荐)
2018/02/03 Javascript
JavaScript设计模式之构造函数模式实例教程
2018/07/02 Javascript
JS实现Cookie读、写、删除操作工具类示例
2018/08/28 Javascript
自定义Vue组件打包、发布到npm及使用教程
2019/05/22 Javascript
详解Nuxt内导航栏的两种实现方式
2020/04/16 Javascript
[55:44]完美世界DOTA2联赛决赛 FTD vs Phoenix 第二场 11.08
2020/11/11 DOTA
python实现dict版图遍历示例
2014/02/19 Python
python 编程之twisted详解及简单实例
2017/01/28 Python
Python时间戳使用和相互转换详解
2017/12/11 Python
python实现蒙特卡罗方法教程
2019/01/28 Python
python 利用文件锁单例执行脚本的方法
2019/02/19 Python
利用python绘制数据曲线图的实现
2020/04/09 Python
Pytorch上下采样函数--interpolate用法
2020/07/07 Python
python根据字典的键来删除元素的方法
2020/08/16 Python
Pycharm学生免费专业版安装教程的方法步骤
2020/09/24 Python
python画图时设置分辨率和画布大小的实现(plt.figure())
2021/01/08 Python
Kenneth Cole官网:纽约时尚优雅品牌
2016/11/14 全球购物
Chicco婴儿用品美国官网:汽车座椅、婴儿推车、高脚椅等
2018/11/05 全球购物
外贸业务员的岗位职责
2013/11/23 职场文书
大学生实习自我鉴定
2013/12/11 职场文书
小学数学教学反思
2014/02/02 职场文书
经典团队口号大全
2014/06/21 职场文书
党员国庆节演讲稿范文2014
2014/09/21 职场文书