码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2020-01-28 14:05:33      阅读:216      评论:0      收藏:0      [点我收藏+]

标签:描述   需要   就会   如何   结构   修改   取出   简单   推荐   

问题描述

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

 

此时有两种解决方法:

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

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

 

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

 

我个人推荐第一种方法。

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

标签:描述   需要   就会   如何   结构   修改   取出   简单   推荐   

原文地址:https://www.cnblogs.com/yongy1030/p/12237806.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!