From 12bcbf9bea7bcd0e80f90a1c90fe129c5c2cfedf Mon Sep 17 00:00:00 2001 From: lilei Date: Tue, 7 Dec 2021 10:04:00 +0800 Subject: [PATCH] modify GroupNorm initializer --- mindspore/nn/layer/normalization.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/mindspore/nn/layer/normalization.py b/mindspore/nn/layer/normalization.py index 5831cb91b6b..9c261355643 100644 --- a/mindspore/nn/layer/normalization.py +++ b/mindspore/nn/layer/normalization.py @@ -1113,14 +1113,10 @@ class GroupNorm(Cell): self.eps = validator.check_value_type('eps', eps, (float,), type(self).__name__) self.affine = validator.check_bool(affine, arg_name="affine", prim_name=self.cls_name) - gamma = initializer(gamma_init, num_channels) - beta = initializer(beta_init, num_channels) - if self.affine: - self.gamma = Parameter(gamma, name='gamma') - self.beta = Parameter(beta, name='beta') - else: - self.gamma = gamma - self.beta = beta + self.gamma = Parameter(initializer( + gamma_init, num_channels), name="gamma", requires_grad=affine) + self.beta = Parameter(initializer( + beta_init, num_channels), name="beta", requires_grad=affine) self.shape = F.shape self.reshape = F.reshape self.reduce_mean = P.ReduceMean(keep_dims=True)