pytorch中修改后的模型如何加載預訓練模型

問題描述

簡單來說，比如你要加載一個vgg16模型，但是你自己需要的網絡結構並不是原本的vgg16網絡，可能你刪掉某些層，可能你改掉某些層，這時你去加載預訓練模型，就會報錯，錯誤原因就是你的模型和原本的模型不匹配。

 

此時有兩種解決方法：

1、重新解析參數的字典，將預訓練模型的參數提取出來，然后放在自己的模型中對應的位置

2、直接用原本的vgg16網絡去加載預訓練模型，然后再修改網絡。

 

具體操作待續吧......

 

我個人推薦第一種方法。