這期內(nèi)容當(dāng)中小編將會(huì)給大家?guī)?lái)有關(guān)如何使用pytorch預(yù)訓(xùn)練層，文章內(nèi)容豐富且以專業(yè)的角度為大家分析和敘述，閱讀完這篇文章希望大家可以有所收獲。

在禹州等地區(qū)，都構(gòu)建了全面的區(qū)域性戰(zhàn)略布局，加強(qiáng)發(fā)展的系統(tǒng)性、市場(chǎng)前瞻性、產(chǎn)品創(chuàng)新能力，以專注、極致的服務(wù)理念，為客戶提供做網(wǎng)站、網(wǎng)站設(shè)計(jì) 網(wǎng)站設(shè)計(jì)制作按需設(shè)計(jì)網(wǎng)站,公司網(wǎng)站建設(shè),企業(yè)網(wǎng)站建設(shè),成都品牌網(wǎng)站建設(shè),成都營(yíng)銷網(wǎng)站建設(shè),成都外貿(mào)網(wǎng)站制作,禹州網(wǎng)站建設(shè)費(fèi)用合理。

pytorch 預(yù)訓(xùn)練層的使用方法

將其他地方訓(xùn)練好的網(wǎng)絡(luò)，用到新的網(wǎng)絡(luò)里面

加載預(yù)訓(xùn)練網(wǎng)絡(luò)

1.原先已經(jīng)訓(xùn)練好一個(gè)網(wǎng)絡(luò) AutoEncoder_FC()

2.首先加載該網(wǎng)絡(luò)，讀取其存儲(chǔ)的參數(shù)

3.設(shè)置一個(gè)參數(shù)集

cnnpre = AutoEncoder_FC()
cnnpre.load_state_dict(torch.load('autoencoder_FC.pkl')['state_dict'])
cnnpre_dict =cnnpre.state_dict()

加載新網(wǎng)絡(luò)

1.設(shè)置新的網(wǎng)絡(luò)

2.設(shè)置新網(wǎng)絡(luò)參數(shù)集

cnn= AutoEncoder()
cnn_dict = cnn.state_dict()

更新新網(wǎng)絡(luò)參數(shù)

1.將兩個(gè)參數(shù)集比對(duì)，存在的網(wǎng)絡(luò)參數(shù)保留

2.使用保留下的參數(shù)更新新網(wǎng)絡(luò)參數(shù)集

3.加載新網(wǎng)絡(luò)參數(shù)集到新網(wǎng)絡(luò)中

cnnpre_dict = {k: v for k, v in cnnpre_dict.items() if k in cnn_dict}
cnn_dict.update(cnnpre_dict)
cnn.load_state_dict(cnn_dict)

上述就是小編為大家分享的如何使用pytorch預(yù)訓(xùn)練層了，如果剛好有類似的疑惑，不妨參照上述分析進(jìn)行理解。如果想知道更多相關(guān)知識(shí)，歡迎關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道。

當(dāng)前題目：如何使用pytorch預(yù)訓(xùn)練層-創(chuàng)新互聯(lián)
網(wǎng)站路徑：http://aaarwkj.com/article6/jdoig.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供網(wǎng)站維護(hù)、網(wǎng)站導(dǎo)航、云服務(wù)器、小程序開(kāi)發(fā)、網(wǎng)頁(yè)設(shè)計(jì)公司、做網(wǎng)站

廣告

聲明：本網(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)