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 相关文章推荐
Python数据类型详解(二)列表
May 08 Python
Python实现判断给定列表是否有重复元素的方法
Apr 11 Python
Python使用pylab库实现绘制直方图功能示例
Jun 01 Python
对Python 3.5拼接列表的新语法详解
Nov 08 Python
python+pyqt5实现24点小游戏
Jan 24 Python
详解Django+uwsgi+Nginx上线最佳实战
Mar 14 Python
使用django的ORM框架按月统计近一年内的数据方法
Jul 18 Python
Python字典常见操作实例小结【定义、添加、删除、遍历】
Oct 25 Python
基于python2.7实现图形密码生成器的实例代码
Nov 05 Python
django商品分类及商品数据建模实例详解
Jan 03 Python
python3 xpath和requests应用详解
Mar 06 Python
python GUI库图形界面开发之PyQt5信号与槽的高级使用技巧装饰器信号与槽详细使用方法与实例
Mar 06 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
dedecms中常见问题修改方法总结
2007/03/21 PHP
一个PHP二维数组排序的函数分享
2014/01/17 PHP
PHP empty函数报错解决办法
2014/03/06 PHP
PHP队列用法实例
2014/11/05 PHP
php+javascript实现的动态显示服务器运行程序进度条功能示例
2017/08/07 PHP
自动更新作用
2006/10/08 Javascript
关于javascript中的parseInt使用技巧
2009/09/03 Javascript
js对象的比较
2011/02/26 Javascript
jquery中文乱码的多种解决方法
2013/06/21 Javascript
jquery实现textarea输入框限制字数的方法
2015/01/15 Javascript
javascript组合使用构造函数模式和原型模式实例
2015/06/04 Javascript
JavaScript每天定时更换皮肤样式的方法
2015/07/01 Javascript
JavaScript访问字符串中单个字符的两种方法
2015/07/03 Javascript
JavaScript数组对象实现增加一个返回随机元素的方法
2015/07/27 Javascript
总结jQuery插件开发中的一些要点
2016/05/16 Javascript
用nodejs的实现原理和搭建服务器(动态)
2016/08/10 NodeJs
JS中split()用法(将字符串按指定符号分割成数组)
2016/10/24 Javascript
基于jQuery制作小图标上下滑动特效
2017/01/18 Javascript
Vue组件通信的几种实现方法
2019/04/25 Javascript
jQuery设置下拉框显示与隐藏效果的方法分析
2019/09/15 jQuery
原生javascript的ajax请求及后台PHP响应操作示例
2020/02/24 Javascript
Python匿名函数/排序函数/过滤函数/映射函数/递归/二分法
2019/06/05 Python
Python PyQt5整理介绍
2020/04/01 Python
如何理解Python中的变量
2020/06/01 Python
css3动画效果抖动解决方法
2018/09/03 HTML / CSS
Html5 FileReader实现即时上传图片功能实例代码
2014/09/01 HTML / CSS
HTML5开发动态音频图的实现
2020/07/02 HTML / CSS
NYX Professional Makeup官方网站:专业彩妆和美容产品
2019/10/29 全球购物
创业计划书六个要素
2013/12/26 职场文书
销售员求职个人的自我评价
2014/02/19 职场文书
社区平安建设汇报材料
2014/08/14 职场文书
小学庆六一活动总结
2014/08/28 职场文书
python数字类型和占位符详情
2022/03/13 Python
docker-compose部署Yapi的方法
2022/04/08 Servers
一篇文章带你掌握SQLite3基本用法
2022/06/14 数据库
Golang gRPC HTTP协议转换示例
2022/06/16 Golang