提交 4924749e 编写于 作者: C cuicheng01 提交者: cuicheng01

fix_celoss

上级 5ad28174
......@@ -57,7 +57,6 @@ class CELoss(nn.Layer):
loss = loss.sum()
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.
先完成此消息的编辑!
想要评论请 注册