Pytorch .pth權(quán)重文件的使用解析
更新時間：2020年02月14日 12:08:10 作者：懂懂懂懂懂懂懂
今天小編就為大家分享一篇Pytorch .pth權(quán)重文件的使用解析，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
pytorch最后的權(quán)重文件是.pth格式的。
經(jīng)常遇到的問題：
進(jìn)行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)文章
repr() 輸出對 Python比較友好，而str()的輸出對用戶比較友好。雖然如此，很多情況下這三者的輸出仍然都是完全一樣的2013-03-03
這篇文章主要介紹了Matlab中如何實現(xiàn)將長字符串換行寫問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-01-01
Python 隨機(jī)生成中文驗證碼的實例代碼，需要的朋友可以參考一下2013-03-03
Pandas之?dāng)?shù)據(jù)追加df.append方式
這篇文章主要介紹了Pandas之?dāng)?shù)據(jù)追加df.append方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-08-08
pycharm 使用心得（五）斷點(diǎn)調(diào)試
PyCharm 作為IDE，斷點(diǎn)調(diào)試是必須有的功能。否則，我們還真不如用純編輯器寫的快。2014-06-06
這篇文章主要介紹了Python如何一行輸入多個數(shù)，并存入列表，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-05-05
python中requests爬去網(wǎng)頁內(nèi)容出現(xiàn)亂碼問題解決方法介紹
這篇文章主要介紹了python中requests爬去網(wǎng)頁內(nèi)容出現(xiàn)亂碼問題解決方法，2017-10-10
SciPy是一個基于NumPy構(gòu)建的Python模塊,它集成了多種數(shù)學(xué)算法和函數(shù),這篇文章主要為大家詳細(xì)介紹了如何使用SciPy庫實現(xiàn)插值,需要的可以了解下2024-03-03
這篇文章主要介紹了pygame實現(xiàn)井字棋之第一步繪制九宮格,文中有非常詳細(xì)的代碼示例,對正在學(xué)習(xí)python的小伙伴們有非常好的幫助,需要的朋友可以參考下2021-05-05
《詐金花》又叫三張牌，是在全國廣泛流傳的一種民間多人紙牌游戲。比如JJ比賽中的詐金花(贏三張)，具有獨(dú)特的比牌規(guī)則。本文江將通過Python語言實現(xiàn)這一游戲，需要的可以參考一下2022-02-02