fix batchnorm bug

6 years ago · a974e65bb6
parent 6d3709ebd1
commit a974e65bb6
1 changed files with 1 additions and 1 deletions
--- a/mindspore/nn/layer/normalization.py
+++ b/mindspore/nn/layer/normalization.py
@ -376,7 +376,7 @@ class GlobalBatchNorm(_BatchNorm):
                                          use_batch_statistics,
                                          group)
        self.group = check_int_positive(group)
-        if self.group <=1:
+        if self.group <= 1:
            raise ValueError("the number of group must be greater than 1.")
    def _check_data_dim(self, x):
        if x.dim == 0: