using new api cross_entropy. (#573)

ZHUI · web-flow · commit 840149dfce57 · 2021-06-16T21:21:02.000+08:00
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):
