using new api cross_entropy. (PaddlePaddle#573)

wangxicoding · Jun 16, 2021 · 840149d · 840149d
1 parent fa1e0ba
commit 840149d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/text_graph/erniesage/models/loss.py b/examples/text_graph/erniesage/models/loss.py
@@ -48,7 +48,7 @@ def __init__(self, config):
         super(SoftmaxWithCrossEntropy, self).__init__()
 
     def forward(self, logits, label):
-        return paddle.mean(F.softmax_with_cross_entropy(logits, label))
+        return F.cross_entropy(logits, label, reduction="mean")
 
 
 class HingeLoss(nn.Layer):