Pytorch .pth權(quán)重文件的使用解析
更新時間：2020年02月14日 12:08:10 作者：懂懂懂懂懂懂懂
今天小編就為大家分享一篇Pytorch .pth權(quán)重文件的使用解析，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
pytorch最后的權(quán)重文件是.pth格式的。
經(jīng)常遇到的問題：
進行finutune時，改配置文件中的學(xué)習(xí)率，發(fā)現(xiàn)程序跑起來后竟然保持了以前的學(xué)習(xí)率， 并沒有使用新的學(xué)習(xí)率。
原因：
首先查看.pth文件中的內(nèi)容，我們發(fā)現(xiàn)它其實是一個字典格式的文件
其中保存了optimizer和scheduler，所以再次加載此文件時會使用之前的學(xué)習(xí)率。
我們只需要權(quán)重，也就是model部分，將其導(dǎo)出就可以了
import torch
original = torch.load('path/to/your/checkpoint.pth')
new = {"model": original["model"]}
torch.save(new, 'path/to/new/checkpoint.pth')
以上這篇Pytorch .pth權(quán)重文件的使用解析就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
python django 實現(xiàn)驗證碼的功能實例代碼
本篇文章主要介紹了python django 實現(xiàn)驗證碼的功能實例代碼，具有一定的參考價值，感興趣的小伙伴們可以參考一下2017-05-05
這篇文章主要介紹了使用Python+Appuim 清理微信,本文通過實例代碼給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-01-01
這篇文章主要介紹了十行代碼實現(xiàn)python向手機推送通知,這里使用的是pushplus的服務(wù)，代碼也很簡單，運行代碼后也是很快就可以收到消息推送，需要的朋友可以參考下2022-04-04
Folium是一個基于leaflet.js的Python地圖庫，其中，Leaflet是一個非常輕的前端地圖可視化庫，本文重點給大家介紹Python實現(xiàn)地圖可視化folium完整過程，感興趣的朋友跟隨小編一起看看吧2021-05-05
同時安裝Python2 & Python3 cmd下版本自由選擇的方法
下面小編就為大家分享一篇同時安裝Python2 & Python3 cmd下版本自由選擇的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2017-12-12
這篇文章主要介紹了Python break語句的作用、使用方法,需要的朋友可以參考下2014-03-03
本文主要介紹了python httpx的具體使用，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-04-04
Python?matplotlib實現(xiàn)折線圖的繪制
Matplotlib作為Python的2D繪圖庫，它以各種硬拷貝格式和跨平臺的交互式環(huán)境生成出版質(zhì)量級別的圖形。本文將利用Matplotlib庫繪制折線圖，感興趣的可以了解一下2022-03-03
python向MySQL數(shù)據(jù)庫插入數(shù)據(jù)的操作方法
這篇文章主要介紹了python向MySQL數(shù)據(jù)庫插入數(shù)據(jù),本文通過實例代碼給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2022-11-11
在本篇文章里我們給大家分享了關(guān)于Python爬取成語接龍類網(wǎng)站的相關(guān)知識點，有需要的朋友們學(xué)習(xí)下。2018-10-10