提交 6714a9d1 编写于 作者: C cuicheng01 提交者: cuicheng01

fix_celoss

上级 3bd78eb1
......@@ -51,7 +51,6 @@ class CELoss(nn.Layer):
loss = paddle.sum(x * label, axis=-1)
else:
if label.shape[-1] == x.shape[-1]:
label = F.softmax(label, axis=-1)
soft_label = True
else:
soft_label = False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册