溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
這期內容當中小編將會給大家帶來有關如何使用pytorch預訓練層，文章內容豐富且以專業的角度為大家分析和敘述，閱讀完這篇文章希望大家可以有所收獲。
pytorch 預訓練層的使用方法
將其他地方訓練好的網絡，用到新的網絡里面
加載預訓練網絡
1.原先已經訓練好一個網絡 AutoEncoder_FC()
2.首先加載該網絡，讀取其存儲的參數
3.設置一個參數集
cnnpre = AutoEncoder_FC()
cnnpre.load_state_dict(torch.load('autoencoder_FC.pkl')['state_dict'])
cnnpre_dict =cnnpre.state_dict()加載新網絡
1.設置新的網絡
2.設置新網絡參數集
cnn= AutoEncoder() cnn_dict = cnn.state_dict()
更新新網絡參數
1.將兩個參數集比對，存在的網絡參數保留
2.使用保留下的參數更新新網絡參數集
3.加載新網絡參數集到新網絡中
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預訓練層了，如果剛好有類似的疑惑，不妨參照上述分析進行理解。如果想知道更多相關知識，歡迎關注億速云行業資訊頻道。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
