如何只加载预训练的vgg16中某些层的参数
收藏
如何只加载vgg16预训练模型中某些层的参数到我自己的模型里,我模型里的层名字和vgg16的不一样,但shape是一样的。
0
收藏
请登录后评论
存储的各层的数据都是以字典方式存储的,可以考虑手动迁移
给你个参考:
这里就resnet18,然后去里面取特定的层。