首页 Paddle框架 帖子详情
paddlepaddle是如何加载预训练模型的?
收藏
快速回复
Paddle框架 问答模型训练 1997 1
paddlepaddle是如何加载预训练模型的?
收藏
快速回复
Paddle框架 问答模型训练 1997 1

如题,例如,我要加载imagenet上预训练的ResNet50,这个预训练模型里面还有最后全连接层(FC层)的参数,但我的网络是不需要的,只需要把前面的卷积层参数加载了,那在调用fluid.io.load_vars时,底层是如何实现的呢?是通过每一层的名称加载的吗?还是它会根据网络结构自动加载,而不管你的新网络各层的名称和预训练模型层的名称是否相同?

0
收藏
回复
全部评论(1)
时间顺序
星光ld1
#2 回复于2020-05

这个不是很确定,我感觉是根据名称加载的,因为之前看到好多写加载权重的网络结构每一层都写了name

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户