zzy34407230
/
mindspore2022
Not watched
1
0
Fork 0
Code
Releases 22 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

!828 Add reduce mean in SoftmaxCrossEntroyWithLogits in Resnet50 example 

Merge pull request !828 from gengdongjie/r0.2
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 6 years ago
parent
8d4511e729 5265c90884
commit
88c32a6f4d
1 changed files with 1 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      example/resnet50_cifar10/train.py

+ 1
- 1
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:


Write Preview
Loading…
Cancel
Save