hyf152
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix batchnorm bug

This commit is contained in:
zhaojichen 2020-04-22 21:41:15 -04:00
parent 68b4524651
commit 9d6ff7ffd0
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
mindspore/nn/layer/normalization.py
View File

@ -366,15 +366,15 @@ class GlobalBatchNorm(_BatchNorm):
use_batch_statistics=True,
group=1):
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)
if self.group <= 1:
raise ValueError("the number of group must be greater than 1.")