首页 Paddle框架 帖子详情
softmax_with_cross_entropy软标签问题
收藏
快速回复
Paddle框架 问答模型训练 1291 4
softmax_with_cross_entropy软标签问题
收藏
快速回复
Paddle框架 问答模型训练 1291 4

ValueError: (InvalidArgument) If Attr(soft_label) == true, the axis dimension of Input(X) and Input(Label) should be equal.
[Hint: Expected logits_dims[axis] == labels_dims[axis], but received logits_dims[axis]:102 != labels_dims[axis]:1.] (at /paddle/paddle/fluid/operators/softmax_with_cross_entropy_op.cc:171)

使用硬标签的交叉熵可以跑起来,但是使用软标签就出现问题,分类数为102,标签维度为1,维度不统一,我是用paddle.vision.DatasetFolder(train_image_dir)读取数据集的。

应该怎么修改能跑起来?谢谢大家!

0
收藏
回复
全部评论(4)
时间顺序
FutureSI
#2 回复于2021-10

这里算loss时加了softmax,模型里最后那层softmax就得去掉了

0
回复
T
Toto-Alfredo
#3 回复于2021-10
这里算loss时加了softmax,模型里最后那层softmax就得去掉了

模型最后的连接层是2048到102,输出的分类数是没错的,应该是这个标签值有问题吧?

0
回复
FutureSI
#4 回复于2021-10

我理解用标签平滑的话,不能直接用hapi了

0
回复
T
Toto-Alfredo
#5 回复于2021-10
我理解用标签平滑的话,不能直接用hapi了

好吧,谢谢了

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