fix globalbatchnorm bug

5 years ago · 0ba35eaec3
parent eb46dd9198
commit 0ba35eaec3
1 changed files with 1 additions and 0 deletions
--- a/mindspore/nn/layer/normalization.py
+++ b/mindspore/nn/layer/normalization.py
@ -117,6 +117,7 @@ class _BatchNorm(Cell):
        return group_list

    def _global_sync(self, x):
+        """calculate global batch normalization output"""
        if len(self.shape(x)) == 4:
            axes = (0, 2, 3)
            re_shape = (1, self.num_features, 1, 1)