Pytorch .pth权重文件的使用解析
更新时间：2020年02月14日 12:08:10 作者：懂懂懂懂懂懂懂
今天小编就为大家分享一篇Pytorch .pth权重文件的使用解析，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
pytorch最后的权重文件是.pth格式的。
经常遇到的问题：
进行finutune时，改配置文件中的学习率，发现程序跑起来后竟然保持了以前的学习率， 并没有使用新的学习率。
原因：
首先查看.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 django 实现验证码的功能实例代码，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2017-05-05
这篇文章主要介绍了使用Python+Appuim 清理微信,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-01-01
这篇文章主要介绍了十行代码实现python向手机推送通知,这里使用的是pushplus的服务，代码也很简单，运行代码后也是很快就可以收到消息推送，需要的朋友可以参考下2022-04-04
Folium是一个基于leaflet.js的Python地图库，其中，Leaflet是一个非常轻的前端地图可视化库，本文重点给大家介绍Python实现地图可视化folium完整过程，感兴趣的朋友跟随小编一起看看吧2021-05-05
同时安装Python2 & Python3 cmd下版本自由选择的方法
下面小编就为大家分享一篇同时安装Python2 & Python3 cmd下版本自由选择的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2017-12-12
这篇文章主要介绍了Python break语句的作用、使用方法,需要的朋友可以参考下2014-03-03
本文主要介绍了python httpx的具体使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-04-04
Matplotlib作为Python的2D绘图库，它以各种硬拷贝格式和跨平台的交互式环境生成出版质量级别的图形。本文将利用Matplotlib库绘制折线图，感兴趣的可以了解一下2022-03-03
这篇文章主要介绍了python向MySQL数据库插入数据,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2022-11-11
在本篇文章里我们给大家分享了关于Python爬取成语接龙类网站的相关知识点，有需要的朋友们学习下。2018-10-10
最新评论