pytorch 預訓練層的使用方法
更新時間：2019年08月20日 09:38:37 作者：zzw小凡
今天小編就為大家分享一篇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 預訓練層的使用方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要給大家介紹了關于Python如何利用正則表達式爬取爬取古詩文網站信息，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-12-12
今天小編就為大家分享一篇利用Python讀取txt文檔的方法講解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-06-06
這篇文章主要介紹了詳解TensorFlow在windows上安裝與簡單示例，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-03-03
今天小編就為大家分享一篇在Django中URL正則表達式匹配的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-12-12
盡管網上搜到許多篇關于pycharm如何安裝pygame庫,但是跟著要求做還是安裝失敗,下面這篇文章主要給大家介紹了關于python中pygame安裝不成功的解決辦法,需要的朋友可以參考下2023-05-05
這篇文章主要介紹了Python入門教程(十五)Python的字典,Python是一門非常強大好用的語言,也有著易上手的特性,本文為入門教程,需要的朋友可以參考下2023-04-04
這篇文章主要介紹了教你使用Python提取視頻中的美女圖片,使用?Python?寫一個逐幀無損保存視頻畫面的小腳本，大概分為三個步驟，詳細過程跟隨小編一起看看吧2022-04-04
Linux下通過python訪問MySQL、Oracle、SQL Server數據庫的方法
這篇文章主要介紹了Linux下通過python訪問MySQL、Oracle、SQL Server數據庫的方法,需要的朋友可以參考下2016-04-04
python利用opencv調用攝像頭實現(xiàn)目標檢測
這篇文章主要為大家介紹了python利用opencv調用攝像頭實現(xiàn)目標檢測的示例代碼，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-05-05
Python+wxPython實現(xiàn)自動生成PPTX文檔程序
這篇文章主要介紹了如何使用 wxPython 模塊和 python-pptx 模塊來編寫一個程序，用于生成包含首頁、內容頁和感謝頁的 PPTX 文檔，感興趣的小伙伴可以學習一下2023-08-08