這篇文章主要介紹了Pytorch加載訓(xùn)練好的模型遇到error問(wèn)題怎么辦，具有一定借鑒價(jià)值，感興趣的朋友可以參考下，希望大家閱讀完這篇文章之后大有收獲，下面讓小編帶著大家一起了解一下。

成都創(chuàng)新互聯(lián)是專業(yè)的曲麻萊網(wǎng)站建設(shè)公司，曲麻萊接單;提供成都網(wǎng)站制作、做網(wǎng)站,網(wǎng)頁(yè)設(shè)計(jì),網(wǎng)站設(shè)計(jì),建網(wǎng)站,PHP網(wǎng)站建設(shè)等專業(yè)做網(wǎng)站服務(wù);采用PHP框架,可快速的進(jìn)行曲麻萊網(wǎng)站開(kāi)發(fā)網(wǎng)頁(yè)制作和功能擴(kuò)展;專業(yè)做搜索引擎喜愛(ài)的網(wǎng)站,專業(yè)的做網(wǎng)站團(tuán)隊(duì),希望更多企業(yè)前來(lái)合作!

pytorch 已經(jīng)非常友好了 保存模型和加載模型都只需要一條簡(jiǎn)單的命令

#保存整個(gè)網(wǎng)絡(luò)和參數(shù)
torch.save(your_net, 'save_name.pkl')
#加載保存的模型
net = torch.load('save_name.pkl')

因?yàn)槲冶容^懶我就想直接把整個(gè)網(wǎng)絡(luò)都保存下來(lái)，然后在test文件中直接load一下不就好了？

就遭受了這樣的錯(cuò)誤。看錯(cuò)了error信息，把‘Net'看成‘net'。報(bào)錯(cuò)沒(méi)有屬性‘net'？這個(gè)不是我自己寫的變量名么？

-----------------瞎搗鼓1h后（呵呵呵）----------------

回頭看error，沒(méi)有屬性‘Net'，Net？？？

我當(dāng)下明白過(guò)來(lái)，應(yīng)該是test文件中沒(méi)有把它import進(jìn)來(lái)，test中就沒(méi)有任何關(guān)于Net的信息。我直接把定義的Net復(fù)制進(jìn)了test.py，就順利加載了訓(xùn)練好的模型。

但是我也有一個(gè)疑問(wèn)，我理解的把整個(gè)模型保存難道不是把它的結(jié)構(gòu)都保存下來(lái)了么？為什么還要再把這個(gè)網(wǎng)絡(luò)import一次？來(lái)自python、pytorch、面向?qū)ο缶幊倘卧“椎囊苫?，先存?zhèn)€疑，搞懂了再來(lái)回答。

接下來(lái)試試只保存網(wǎng)絡(luò)參數(shù)

#只保存網(wǎng)絡(luò)參數(shù)
torch.save(your_net.state_dict(), 'save_name.pkl')
#加載保存的模型
net.load_state_dict(torch.load('save_name.pkl'))

保存網(wǎng)絡(luò)參數(shù)

重新定義網(wǎng)絡(luò)

報(bào)錯(cuò)

想死。。。

仔細(xì)看了報(bào)錯(cuò)信息，以我小白的理解，我感覺(jué)保存下來(lái)的可能只是單純的數(shù)據(jù)，而不是一個(gè)對(duì)象（沒(méi)有方法可以操作），或者該對(duì)象沒(méi)有.copy()方法，所以沒(méi)有辦法進(jìn)行.copy()，那肯定是保存哪里出錯(cuò)了。然后發(fā)現(xiàn)保存部分代碼寫錯(cuò)了，改成

print一下 net.state_dict和net.state_dict()，前者輸出的是網(wǎng)絡(luò)結(jié)構(gòu)，后者才是網(wǎng)絡(luò)的參數(shù)。

試著回答之前的問(wèn)題，第二種保存模型的方法只保存了網(wǎng)絡(luò)的參數(shù)（包括卷積層和全連接層每次的weight,bias），所以再加載模型的時(shí)候需要先定義網(wǎng)絡(luò)無(wú)可厚非，就像訓(xùn)練時(shí)候定義網(wǎng)絡(luò)那樣定義就可以；而第一種保存整個(gè)網(wǎng)絡(luò)的方法，保存了一個(gè)網(wǎng)絡(luò)的實(shí)例（包括它的所有結(jié)構(gòu)和參數(shù)），net是Net的一個(gè)實(shí)例，那為什么還要有Class Net的定義呢，還是回答不了。。

那就繼續(xù)存疑，保持探究精神吧。。

感謝你能夠認(rèn)真閱讀完這篇文章，希望小編分享的“Pytorch加載訓(xùn)練好的模型遇到error問(wèn)題怎么辦”這篇文章對(duì)大家有幫助，同時(shí)也希望大家多多支持創(chuàng)新互聯(lián)，關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道，更多相關(guān)知識(shí)等著你來(lái)學(xué)習(xí)!

分享題目：Pytorch加載訓(xùn)練好的模型遇到error問(wèn)題怎么辦-創(chuàng)新互聯(lián)
網(wǎng)站地址：http://chinadenli.net/article2/cdpdic.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供建站公司、網(wǎng)站設(shè)計(jì)、做網(wǎng)站、移動(dòng)網(wǎng)站建設(shè)、品牌網(wǎng)站建設(shè)、ChatGPT

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請(qǐng)盡快告知，我們將會(huì)在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如需處理請(qǐng)聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時(shí)需注明來(lái)源： 創(chuàng)新互聯(lián)