解决pytorch 保存模型遇到的问题
更新时间：2021年03月03日 11:15:08 作者：Zawo1995
这篇文章主要介绍了解决pytorch 保存模型遇到的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
今天用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中的lambda表达式的用法,包括其与def之间的区别,需要的朋友可以参考下2015-08-08
GUI图形用户界面编程,我们可以通过python提供的丰富的组件,快速的实现使用图形的界面和用户交互, GUI编程类似于“搭积⽊”,将⼀个个组件(Widget)放到窗⼝中,这篇文章主要给大家介绍了基于Python的GUI图形用户界面编程的相关资料,需要的朋友可以参考下2022-12-12
这篇文章主要介绍了Python监听键盘和鼠标事件的示例代码，帮助大家更好的理解和使用python，提高办公效率，感兴趣的朋友可以了解下2020-11-11
这篇文章主要介绍了Python实现matplotlib显示中文的方法,结合实例形式详细总结分析了Python使用matplotlib库绘图时显示中文的相关操作技巧与注意事项,需要的朋友可以参考下2018-02-02
这篇文章主要介绍了Python Thread虚假唤醒概念与防范，虚假唤醒是一种现象，它只会出现在多线程环境中，指的是在多线程环境下，多个线程等待在同一个条件上，等到条件满足时，所有等待的线程都被唤醒，但由于多个线程执行的顺序不同2023-02-02
Appium+python自动化之连接模拟器并启动淘宝APP（超详解）
这篇文章主要介绍了Appium+python自动化之 连接模拟器并启动淘宝APP（超详解）本文以淘宝app为例，通过实例代码给大家介绍的非常详细,需要的朋友可以参考下2019-06-06
Python使用cx_Oracle模块将oracle中数据导出到csv文件的方法
这篇文章主要介绍了Python使用cx_Oracle模块将oracle中数据导出到csv文件的方法,涉及Python中cx_Oracle模块与csv模块操作Oracle数据库及csv文件的相关技巧,需要的朋友可以参考下2015-05-05
在大家的日常python程序的编写过程中，都会有自己解决某个问题的解决办法，或者是在程序的调试过程中，用来帮助调试的程序公式，本文总结了22个万用公式，感兴趣的可以了解一下2021-07-07
matlab和Excel的数据交互操作(非xlsread和xlswrite)
在使用MATLAB时,可能会遇到很多表格数据的处理,有时MATLAB也需要利用现存的表格数据实现操作目的,下面这篇文章主要给大家介绍了关于matlab和Excel的交互操作的相关资料,非xlsread和xlswrite,需要的朋友可以参考下2021-08-08
本文主要给大家讲解如何使用pyqt5打开摄像头并在label上显示，以及优化成圆形label框。通过实例代码给大家介绍的非常详细，感兴趣的朋友跟随小编一起看看吧2022-01-01
最新评论