pytorch中修改后的模型如何加载预训练模型

Posted 2020-11-25 yongy1030

问题描述

简单来说，比如你要加载一个vgg16模型，但是你自己需要的网络结构并不是原本的vgg16网络，可能你删掉某些层，可能你改掉某些层，这时你去加载预训练模型，就会报错，错误原因就是你的模型和原本的模型不匹配。

 

此时有两种解决方法：

1、重新解析参数的字典，将预训练模型的参数提取出来，然后放在自己的模型中对应的位置

2、直接用原本的vgg16网络去加载预训练模型，然后再修改网络。

 

具体操作待续吧......

 

我个人推荐第一种方法。