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