问题描述
简单来说,比如你要加载一个vgg16模型,但是你自己需要的网络结构并不是原本的vgg16网络,可能你删掉某些层,可能你改掉某些层,这时你去加载预训练模型,就会报错,错误原因就是你的模型和原本的模型不匹配。
此时有两种解决方法:
1、重新解析参数的字典,将预训练模型的参数提取出来,然后放在自己的模型中对应的位置
2、直接用原本的vgg16网络去加载预训练模型,然后再修改网络。
具体操作待续吧......
我个人推荐第一种方法。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:pytorch中修改后的模型如何加载预训练模型 - Python技术站