yankf21
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!828 Add reduce mean in SoftmaxCrossEntroyWithLogits in Resnet50 example 

Merge pull request !828 from gengdongjie/r0.2
This commit is contained in:
mindspore-ci-bot 2020-04-29 08:44:01 +08:00 committed by Gitee
parent 8d4511e729 5265c90884
commit 88c32a6f4d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
example/resnet50_cifar10/train.py
View File

@ -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: