fix globalbatchnorm bug

5 years ago · 6c9a54afa1
parent 8261cfd019
commit 6c9a54afa1
1 changed files with 1 additions and 1 deletions
--- a/mindspore/nn/layer/normalization.py
+++ b/mindspore/nn/layer/normalization.py
@ -119,7 +119,7 @@ class _BatchNorm(Cell):
    def _shape_infer(self, x):
        """global batch normalization shape and axes infer"""
        if len(self.shape(x)) == 4:
-            axes = (0,2,3)
+            axes = (0, 2, 3)
            re_shape = (1, self.num_features, 1, 1)
        else:
            axes = (0,)