Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!7972 Fix batchnorm infer 

Merge pull request !7972 from VectorSL/master
This commit is contained in:
mindspore-ci-bot 2020-10-30 18:08:55 +08:00 committed by Gitee
parent 6ebb675212 423b4ef926
commit 9b6530e3b3
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/nn/layer/normalization.py
View File

@ -108,7 +108,7 @@ class _BatchNorm(Cell):
self.bn_train = P.FusedBatchNorm(mode=1,
epsilon=self.eps,
momentum=self.momentum)
self.bn_infer = P.BatchNorm(is_training=False, epsilon=self.eps)
self.bn_infer = P.BatchNorm(is_training=False, epsilon=self.eps, data_format=self.format)
self.enable_global_sync = self.is_global and (self.is_ge_backend or (self.is_graph_mode and self.is_ascend))
self.enable_default_train = self.is_graph_mode and not self.is_global and \
(self.is_ge_backend or self.is_ascend)

2
model_zoo/official/cv/resnet/README.md
View File

@ -513,7 +513,7 @@ result: {'top_5_accuracy': 0.9342589628681178, 'top_1_accuracy': 0.7680657810499
| Uploaded Date | 04/01/2020 (month/day/year) | 08/01/2020 (month/day/year) |
| MindSpore Version | 0.1.0-alpha | 0.6.0-alpha |
| Dataset | ImageNet2012 | ImageNet2012 |
| batch_size | 256 | 32 |
| batch_size | 256 | 256 |
| outputs | probability | probability |
| Accuracy | 76.70% | 76.74% |
| Model for inference | 98M (.air file) | |