基于pytorch的保存和加載模型參數(shù)的方法
當(dāng)我們花費(fèi)大量的精力訓(xùn)練完網(wǎng)絡(luò)，下次預(yù)測(cè)數(shù)據(jù)時(shí)不想再（有時(shí)也不必再）訓(xùn)練一次時(shí)，這時(shí)候torch.save(),torch.load()就要登場(chǎng)了。
保存和加載模型參數(shù)有兩種方式：
方式一：
torch.save(net.state_dict(),path):
功能：保存訓(xùn)練完的網(wǎng)絡(luò)的各層參數(shù)（即weights和bias)
其中：net.state_dict()獲取各層參數(shù)，path是文件存放路徑(通常保存文件格式為.pt或.pth)
net2.load_state_dict(torch.load(path)):
功能：加載保存到path中的各層參數(shù)到神經(jīng)網(wǎng)絡(luò)
注意：不可以直接為torch.load_state_dict(path)，此函數(shù)不能直接接收字符串類型參數(shù)
方式二：
torch.save(net,path):
功能：保存訓(xùn)練完的整個(gè)網(wǎng)絡(luò)模型（不止weights和bias）
net2=torch.load(path):
功能：加載保存到path中的整個(gè)神經(jīng)網(wǎng)絡(luò)
說(shuō)明：官方推薦方式一，原因自然是保存的內(nèi)容少，速度會(huì)更快。
以上這篇基于pytorch的保存和加載模型參數(shù)的方法就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了python中封裝token問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-12-12
python數(shù)據(jù)結(jié)構(gòu)之鏈表的實(shí)例講解
下面小編就為大家?guī)?lái)一篇python數(shù)據(jù)結(jié)構(gòu)之鏈表的實(shí)例講解。小編覺(jué)得挺不錯(cuò)的，現(xiàn)在就分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2017-07-07
python3報(bào)錯(cuò)check_hostname?requires?server_hostname的解決
這篇文章主要介紹了python3報(bào)錯(cuò)check_hostname?requires?server_hostname的解決,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2023-12-12
Python優(yōu)秀開(kāi)源項(xiàng)目Rich源碼解析的流程分析
這篇文章主要介紹了Python優(yōu)秀開(kāi)源項(xiàng)目Rich源碼解析,本文通過(guò)實(shí)例圖文相結(jié)合給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2020-07-07
Python搭建代理IP池實(shí)現(xiàn)接口設(shè)置與整體調(diào)度
這篇文章主要介紹了Python搭建代理IP池實(shí)現(xiàn)接口設(shè)置與整體調(diào)度，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-10-10
Python數(shù)據(jù)處理-導(dǎo)入導(dǎo)出excel數(shù)據(jù)
這篇文章主要介紹了Python數(shù)據(jù)處理-導(dǎo)入導(dǎo)出excel數(shù)據(jù)，Python的一大應(yīng)用就是數(shù)據(jù)分析了，而數(shù)據(jù)分析中，經(jīng)常碰到需要處理Excel數(shù)據(jù)的情況。這里做一個(gè)Python處理Excel數(shù)據(jù)的總結(jié),需要的小伙伴可以參考一下2022-01-01
這篇文章主要為大家詳細(xì)介紹了python遍歷文件目錄、批量處理同類文件，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2019-08-08
這篇文章主要介紹了Python中random模塊用法,實(shí)例分析了Python中random模塊的使用技巧及字符串操作相關(guān)方法,需要的朋友可以參考下2015-05-05
這篇文章主要介紹了Python進(jìn)階:生成器 懶人版本的迭代器詳解,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-06-06
這篇文章主要介紹了深入了解Python iter() 方法的知識(shí),文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-07-07