pytorch 預訓練層的使用方法

pytorch 預訓練層的使用方法

將其他地方訓練好的網路程式設計客棧，用到新的網路裡面

載入預訓練網路

1.原先已經訓練好乙個網路 autoencoder_fc()

2.首先載入該網路，讀取其儲存的引數

3.設定乙個引數集

cnnpre = autoencoder_fc()

cnnpre.lo程式設計客棧ad_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 =

cnn_dict.update(cnnpre_dict)

cnn.load_state_dict(cnn_dict)

本文標題: pytorch 預訓練層的使用方法

本文位址: /jiaoben/python/268898.html

pytorch 使用預訓練層

將其他地方訓練好的網路，用到新的網路裡面 1.原先已經訓練好乙個網路 autoencoder fc 2.首先載入該網路，讀取其儲存的引數 3.設定乙個引數集 cnnpre autoencoder fc cnnpre.load state dict torch.load autoencoder fc....

pytorch載入預訓練模型後，訓練指定層

1 有了已經訓練好的模型引數，對這個模型的某些層做了改變，如何利用這些訓練好的模型引數繼續訓練 pretrained params torch.load pretrained model model the new model model.load state dict pretrained par...

pytorch載入預訓練模型後，訓練指定層

1 有了已經訓練好的模型引數，對這個模型的某些層做了改變，如何利用這些訓練好的模型引數繼續訓練 pretrained params torch.load pretrained model model the new model model.load state dict pretrained par...