pytorch模型存儲(chǔ)的2種實(shí)現(xiàn)方法
更新時(shí)間：2020年02月14日 12:00:56 作者：慢行厚積
今天小編就為大家分享一篇pytorch模型存儲(chǔ)的2種實(shí)現(xiàn)方法，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧
1、保存整個(gè)網(wǎng)絡(luò)結(jié)構(gòu)信息和模型參數(shù)信息：
torch.save(model_object, './model.pth')
直接加載即可使用：
model = torch.load('./model.pth')
2、只保存網(wǎng)絡(luò)的模型參數(shù)-推薦使用
torch.save(model_object.state_dict(), './params.pth')
加載則要先從本地網(wǎng)絡(luò)模塊導(dǎo)入網(wǎng)絡(luò)，然后再加載參數(shù)：
from models import AgeModel
model = AgeModel()
model.load_state_dict(torch.load('./params.pth'))
以上這篇pytorch模型存儲(chǔ)的2種實(shí)現(xiàn)方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python中使用Matplotlib進(jìn)行多圖繪制的詳細(xì)教程
Matplotlib是Python中強(qiáng)大的數(shù)據(jù)可視化工具,支持繪制多圖,這篇文章主要介紹了Python中使用Matplotlib進(jìn)行多圖繪制的相關(guān)資料,包括使用subplot、subplots和GridSpec進(jìn)行布局,并提供了調(diào)整圖表樣式和布局的方法,需要的朋友可以參考下2024-11-11
在python中我們偶爾會(huì)用到輸出不換行的效果，python2中使用逗號，即可，而python3中使用end=''來實(shí)現(xiàn)的，這里簡單為大家介紹一下,需要的朋友可以參考下2017-11-11
本文主要介紹了python里面單雙下劃線的區(qū)別詳解，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-04-04
Python調(diào)用DeepSeek?API的完整操作指南
本文將詳細(xì)介紹如何使用?Python?調(diào)用?DeepSeek?API,實(shí)現(xiàn)流式對話并保存對話記錄,相比?Go?版本,Python?實(shí)現(xiàn)更加簡潔優(yōu)雅,適合快速開發(fā)和原型驗(yàn)證,文中通過代碼示例講解的非常詳細(xì),需要的朋友可以參考下2025-02-02
這篇文章主要介紹了Python subprocess庫的使用詳解，小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧2018-10-10
裝飾器的運(yùn)用是Python編程中的一項(xiàng)高級技巧,這里我們由淺入深,整理了12步入門Python中的decorator裝飾器使用方法,需要的朋友可以參考下2016-06-06
大家好，本篇文章主要講的是Python進(jìn)制轉(zhuǎn)換用法詳解，感興趣的同學(xué)趕快來看一看吧，對你有幫助的話記得收藏一下2022-01-01
Python中的內(nèi)置函數(shù)isdigit()
這篇文章主要介紹了Python中的內(nèi)置函數(shù)isdigit()，具有很好的參考價(jià)值，希望對大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-11-11
Python中實(shí)現(xiàn)定時(shí)任務(wù)常見的幾種方式
在Python中,實(shí)現(xiàn)定時(shí)任務(wù)是一個(gè)常見的需求,無論是在自動(dòng)化腳本、數(shù)據(jù)處理、系統(tǒng)監(jiān)控還是其他許多應(yīng)用場景中,Python提供了多種方法來實(shí)現(xiàn)定時(shí)任務(wù),包括使用標(biāo)準(zhǔn)庫、第三方庫以及系統(tǒng)級別的工具,本文將詳細(xì)介紹幾種常見的Python定時(shí)任務(wù)實(shí)現(xiàn)方式2024-08-08
這篇文章主要為大家介紹了python調(diào)用matlab，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2021-11-11