Pytorch .pth权重文件的使用解析


Posted in Python onFebruary 14, 2020

pytorch最后的权重文件是.pth格式的。

经常遇到的问题:

进行finutune时,改配置文件中的学习率,发现程序跑起来后竟然保持了以前的学习率, 并没有使用新的学习率。

原因:

首先查看.pth文件中的内容,我们发现它其实是一个字典格式的文件

Pytorch .pth权重文件的使用解析

其中保存了optimizer和scheduler,所以再次加载此文件时会使用之前的学习率。

我们只需要权重,也就是model部分,将其导出就可以了

import torch
 
original = torch.load('path/to/your/checkpoint.pth')
 
new = {"model": original["model"]}
torch.save(new, 'path/to/new/checkpoint.pth')

以上这篇Pytorch .pth权重文件的使用解析就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
git使用.gitignore设置不生效或不起作用问题的解决方法
Jun 01 Python
Python编程实现的图片识别功能示例
Aug 03 Python
对变量赋值的理解--Pyton中让两个值互换的实现方法
Nov 29 Python
python PyTorch参数初始化和Finetune
Feb 11 Python
Python KMeans聚类问题分析
Feb 23 Python
python实现textrank关键词提取
Jun 22 Python
Pandas之MultiIndex对象的示例详解
Jun 25 Python
python SQLAlchemy的Mapping与Declarative详解
Jul 04 Python
python实现桌面托盘气泡提示
Jul 29 Python
调用其他python脚本文件里面的类和方法过程解析
Nov 15 Python
python函数中将变量名转换成字符串实例
May 11 Python
python游戏开发之pygame实现接球小游戏
Apr 22 Python
Python netmiko模块的使用
Feb 14 #Python
pytorch模型存储的2种实现方法
Feb 14 #Python
Python3运算符常见用法分析
Feb 14 #Python
pytorch实现从本地加载 .pth 格式模型
Feb 14 #Python
python 实现线程之间的通信示例
Feb 14 #Python
Python3变量与基本数据类型用法实例分析
Feb 14 #Python
OpenCV中VideoCapture类的使用详解
Feb 14 #Python
You might like
Apache, PHP在Windows 9x/NT下的安装与配置 (一)
2006/10/09 PHP
PHP 远程文件管理,可以给表格排序,遍历目录,时间排序
2009/08/07 PHP
php笔记之:文章中图片处理的使用
2013/04/26 PHP
php自定义类fsocket模拟post或get请求的方法
2015/07/31 PHP
PHP实现生成唯一会员卡号
2015/08/24 PHP
微信公众平台DEMO(PHP)
2016/05/04 PHP
js格式化时间小结
2014/11/03 Javascript
基于jquery实现即时检查格式是否正确的表单
2016/05/06 Javascript
Angularjs的ng-repeat中去除重复数据的方法
2016/08/05 Javascript
Actionscript与javascript交互实例程序(修改)
2016/09/22 Javascript
js实现表单提交后不重新刷新当前页面
2016/11/30 Javascript
ES6 javascript的异步操作实例详解
2017/10/30 Javascript
利用nodeJs anywhere搭建本地服务器环境的方法
2018/05/12 NodeJs
bootstrap自定义样式之bootstrap实现侧边导航栏功能
2018/09/10 Javascript
javascript数组常见操作方法实例总结【连接、添加、删除、去重、排序等】
2019/06/13 Javascript
jQuery实现判断滚动条滚动到document底部的方法分析
2019/08/27 jQuery
Vue+Element-UI实现上传图片并压缩
2019/11/26 Javascript
JS异步宏队列与微队列原理区别详解
2020/07/02 Javascript
详解JavaScript 事件流
2020/09/02 Javascript
python使用mailbox打印电子邮件的方法
2015/04/30 Python
用Python操作字符串之rindex()方法的使用
2015/05/19 Python
Python中random模块用法实例分析
2015/05/19 Python
python 获取当天每个准点时间戳的实例
2018/05/22 Python
关于tensorflow的几种参数初始化方法小结
2020/01/04 Python
django有外键关系的两张表如何相互查找
2020/02/10 Python
Python实现封装打包自己写的代码,被python import
2020/07/12 Python
Python子进程subpocess原理及用法解析
2020/07/16 Python
使用html5制作loading图的示例
2014/04/14 HTML / CSS
荷兰牛仔裤网上商店:Jeans Centre
2018/04/03 全球购物
英国助听器购物网站:Hearing Direct
2018/08/21 全球购物
德国高端单身人士交友网站:ElitePartner
2018/12/02 全球购物
高考自主招生自荐信
2013/10/20 职场文书
客房部经理岗位职责
2015/02/02 职场文书
毕业论文答辩开场白
2015/05/27 职场文书
Python中使用Lambda函数的5种用法
2021/04/01 Python
JS实现九宫格拼图游戏
2022/06/28 Javascript