Browse Source

fix batchnorm bug

tags/v0.2.0-alpha
zhaojichen 5 years ago
parent
commit
a974e65bb6
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/layer/normalization.py

+ 1
- 1
mindspore/nn/layer/normalization.py View File

@@ -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:


Loading…
Cancel
Save