首页 Paddle框架 帖子详情
使用Resnet50预训练模型调整了输出类别数后,权重形状不一致如何解决
收藏
快速回复
Paddle框架 问答模型训练 99 0
使用Resnet50预训练模型调整了输出类别数后,权重形状不一致如何解决
收藏
快速回复
Paddle框架 问答模型训练 99 0

 

model = paddle.vision.models.resnet50(pretrained=True, num_classes=12)

出现错误

fc.weight receives a shape [2048, 1000], but the expected shape is [2048, 12],

模型输出更新为12时框架内核没有自动将预训练模型输出层权重形状1000转变为12

查了一下API对num_classes=修改,是否会自动初始化最后的全连接层权重没有说明?

如果想删除最后全连接层 num_classes=0,自己增加一个fc,显示RESNET,没有fc属性

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