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

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

pytorch 預訓練層的使用方法

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

加載預訓練網(wǎng)絡

1.原先已經(jīng)訓練好一個網(wǎng)絡 AutoEncoder_FC()

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

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

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

加載新網(wǎng)絡

1.設置新的網(wǎng)絡

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

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

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

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

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

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

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預訓練層了，如果剛好有類似的疑惑，不妨參照上述分析進行理解。如果想知道更多相關(guān)知識，歡迎關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道。

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

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

廣告

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