解决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实现向ppt文件里插入新幻灯片页面的方法
Apr 28 Python
详解Python的Django框架中的通用视图
May 04 Python
python比较两个列表是否相等的方法
Jul 28 Python
python with提前退出遇到的坑与解决方案
Jan 05 Python
numpy中实现二维数组按照某列、某行排序的方法
Apr 04 Python
CentOS7下python3.7.0安装教程
Jul 30 Python
Python SQL查询并生成json文件操作示例
Aug 17 Python
python多进程控制学习小结
Oct 31 Python
python 经典数字滤波实例
Dec 16 Python
关于Tensorflow分布式并行策略
Feb 03 Python
Python如何访问字符串中的值
Feb 09 Python
利用Python实现最小二乘法与梯度下降算法
Feb 21 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编程函数安全篇
2013/01/08 PHP
php文件上传的例子及参数详解
2013/12/12 PHP
PHP实现货币换算的方法
2014/11/29 PHP
PHP实现自动识别Restful API的返回内容类型
2015/02/07 PHP
php生成验证码,缩略图及水印图的类分享
2016/04/07 PHP
PHP与JavaScript针对Cookie的读写、交互操作方法详解
2017/08/07 PHP
PHP实现无限极分类的两种方式示例【递归和引用方式】
2019/03/25 PHP
js的表单操作 简单计算器
2011/12/29 Javascript
JS获取后台Cookies值的小例子
2013/03/04 Javascript
利用javaScript实现点击输入框弹出窗体选择信息
2013/12/11 Javascript
jquery实现勾选复选框触发事件给input赋值
2015/02/01 Javascript
JQuery中Text方法用法实例分析
2015/05/18 Javascript
JavaScript中的数组遍历forEach()与map()方法以及兼容写法介绍
2016/05/19 Javascript
JS面试题大坑之隐式类型转换实例代码
2018/10/14 Javascript
JavaScript作用域链实例详解
2019/01/21 Javascript
js实现随机div颜色位置 类似满天星效果
2019/10/24 Javascript
使用Python进行二进制文件读写的简单方法(推荐)
2016/09/12 Python
pandas的唯一值、值计数以及成员资格的示例
2018/07/25 Python
Django文件存储 自己定制存储系统解析
2019/08/02 Python
Python测试模块doctest使用解析
2019/08/10 Python
Pandas实现DataFrame按行求百分数(比例数)
2019/12/27 Python
Scrapy框架实现的登录网站操作示例
2020/02/06 Python
Python拼接字符串的7种方式详解
2020/03/19 Python
Python定时从Mysql提取数据存入Redis的实现
2020/05/03 Python
Pycharm2020.1安装无法启动问题即设置中文插件的方法
2020/08/07 Python
Selenium webdriver添加cookie实现过程详解
2020/08/12 Python
Python使用windows设置定时执行脚本
2020/11/12 Python
AVIS安飞士奥地利租车官网:提供奥地利、欧洲和全世界汽车租赁
2016/11/29 全球购物
俄罗斯品牌服装在线商店:VIPAVENUE
2020/08/10 全球购物
介绍一下Prototype的$()函数,$F()函数,$A()函数都是什么作用?
2014/03/05 面试题
小学英语课后反思
2014/04/26 职场文书
计算机网络专业自荐信
2014/07/04 职场文书
村干部任职承诺书
2015/01/21 职场文书
颐和园导游词
2015/01/30 职场文书
《酸的和甜的》教学反思
2016/02/18 职场文书
HTML CSS 一个标签实现带动画的抖音LOGO
2022/04/26 HTML / CSS