We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f3f1ac8 commit 2006b6cCopy full SHA for 2006b6c
tensorlayer/cost/mindspore_cost.py
@@ -51,7 +51,7 @@ def softmax_cross_entropy_with_logits(output, target):
51
52
"""
53
54
- outputs = nn.SoftmaxCrossEntropyWithLogits(sparse=True)(output, target)
+ outputs = nn.SoftmaxCrossEntropyWithLogits(sparse=True, reduction='mean')(output, target)
55
return outputs
56
57
0 commit comments