zhaojichen
5 years ago
1 changed files with 9 additions and 9 deletions
Unified View
Diff Options
+ 9
- 9
mindspore/nn/layer/normalization.py
View File
| @@ -366,15 +366,15 @@ class GlobalBatchNorm(_BatchNorm): | |||||
| use_batch_statistics=True, | use_batch_statistics=True, | ||||
| group=1): | group=1): | ||||
| super(GlobalBatchNorm, self).__init__(num_features, | super(GlobalBatchNorm, self).__init__(num_features, | ||||
| eps, | |||||
| momentum, | |||||
| affine, | |||||
| gamma_init, | |||||
| beta_init, | |||||
| moving_mean_init, | |||||
| moving_var_init, | |||||
| use_batch_statistics, | |||||
| group) | |||||
| eps, | |||||
| momentum, | |||||
| affine, | |||||
| gamma_init, | |||||
| beta_init, | |||||
| moving_mean_init, | |||||
| moving_var_init, | |||||
| use_batch_statistics, | |||||
| group) | |||||
| self.group = check_int_positive(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.") | raise ValueError("the number of group must be greater than 1.") | ||||