2020年1月28日
pytorch中修改后的模型如何加载预训练模型
摘要： 问题描述 简单来说，比如你要加载一个vgg16模型，但是你自己需要的网络结构并不是原本的vgg16网络，可能你删掉某些层，可能你改掉某些层，这时你去加载预训练模型，就会报错，错误原因就是你的模型和原本的模型不匹配。 此时有两种解决方法： 1、重新解析参数的字典，将预训练模型的参数提取出来，然后放在自 阅读全文
posted @ 2020-01-28 12:44 立冬以东 阅读(2314) 评论(1) 推荐(0) 编辑