!828 Add reduce mean in SoftmaxCrossEntroyWithLogits in Resnet50 example

Merge pull request !828 from gengdongjie/r0.2

!828 Add reduce mean in SoftmaxCrossEntroyWithLogits in Resnet50 example
Merge pull request !828 from gengdongjie/r0.2
88c32a6f · mindspore-ci-bot · Gitee · 8d4511e7 · 5265c908 · 88c32a6f
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

example/resnet50_cifar10/train.py example/resnet50_cifar10/train.py +1 -1

未找到文件。
--- a/example/resnet50_cifar10/train.py
+++ b/example/resnet50_cifar10/train.py
@@ -68,7 +68,7 @@ if __name__ == '__main__':

    epoch_size = config.epoch_size
    net = resnet50(class_num=config.class_num)
-    loss = SoftmaxCrossEntropyWithLogits(sparse=True)
+    loss = SoftmaxCrossEntropyWithLogits(sparse=True, reduction="mean")


    if args_opt.do_train: