pytorch模型存儲的2種實現方法
更新時間：2020年02月14日 12:00:56 作者：慢行厚積
今天小編就為大家分享一篇pytorch模型存儲的2種實現方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
1、保存整個網絡結構信息和模型參數信息：
torch.save(model_object, './model.pth')
直接加載即可使用：
model = torch.load('./model.pth')
2、只保存網絡的模型參數-推薦使用
torch.save(model_object.state_dict(), './params.pth')
加載則要先從本地網絡模塊導入網絡，然后再加載參數：
from models import AgeModel
model = AgeModel()
model.load_state_dict(torch.load('./params.pth'))
以上這篇pytorch模型存儲的2種實現方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
淺談Python中的可迭代對象、迭代器、For循環(huán)工作機制、生成器
這篇文章主要介紹了Python中的可迭代對象、迭代器、For循環(huán)工作機制、生成器，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-03-03
今天小編就為大家分享一篇Python實現釘釘發(fā)送報警消息的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-02-02
這篇文章主要為大家介紹了python使用TCP實現的對話客戶端和服務器實現示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2023-03-03
這篇文章主要介紹了Python命令行click參數用法解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-12-12
這篇文章主要介紹了Python實現字符串與數組相互轉換功能,結合具體實例形式分析了Python字符串與數組相關轉換功能的相關實現技巧與注意事項,需要的朋友可以參考下2017-09-09
這篇文章主要介紹了Python線程條件變量Condition原理解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-01-01
本文給大家講解的是使用Python的Testlink實現將實現將xml導入至excel表格中，方法非常的簡單，另外附上其他小伙伴的方法，有需要的童鞋們可以參考下。2015-11-11
這篇文章主要為大家詳細介紹了python的數據類型，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下希望能夠給你帶來幫助2022-01-01
這篇文章主要介紹了使用Python發(fā)送各種形式的郵件的方法匯總,包括HTML形式的或者帶圖片以及帶附件的郵件等等,需要的朋友可以參考下2015-11-11
python標準庫turtle海龜繪圖實現簡單奧運五環(huán)
這篇文章主要為大家介紹了python使用turtle實現最簡單簡單奧運五環(huán)繪圖，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-05-05