溫馨提示×
您好，登錄后才能下訂單哦！
點(diǎn)擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》
您好，登錄后才能下訂單哦！
小編給大家分享一下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)它其實(shí)是一個字典格式的文件
其中保存了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)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對大家有所幫助，如果還想學(xué)習(xí)更多知識，歡迎關(guān)注億速云行業(yè)資訊頻道！
免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場，如果涉及侵權(quán)請聯(lián)系站長郵箱：is@yisu.com進(jìn)行舉報，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。
億速云公眾號
手機(jī)網(wǎng)站二維碼