softmax_with_cross_entropy软标签问题
收藏
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
收藏
请登录后评论
这里算loss时加了softmax,模型里最后那层softmax就得去掉了
模型最后的连接层是2048到102,输出的分类数是没错的,应该是这个标签值有问题吧?
我理解用标签平滑的话,不能直接用hapi了
好吧,谢谢了