提交 f2df33e3 编写于 作者: H HydrogenSulfate 提交者: chajchaj

Update loss.py

上级 0bf32484
......@@ -1620,9 +1620,9 @@ def cross_entropy(input,
labels = paddle.uniform(shape, dtype='float64', min=0.1, max=1.0)
labels /= paddle.sum(labels, axis=axis, keepdim=True)
paddle_loss_mean = paddle.nn.functional.cross_entropy(
logits,
labels,
soft_label=True,
logits,
labels,
soft_label=True,
axis=axis,
weight=weight,
reduction=reduction)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册