!358 Fix optimizer.decay_weight bug

Merge pull request !358 from fanglei/master
2020-04-15 18:48:57 +08:00 · 2020-04-15 18:48:57 +08:00 · c241b19a53
parent aa250749cb c4d0bb266a
commit c241b19a53
1 changed files with 1 additions and 1 deletions
--- a/mindspore/nn/optim/optimizer.py
+++ b/mindspore/nn/optim/optimizer.py
@ -109,7 +109,7 @@ class Optimizer(Cell):
            tuple[Tensor], The gradients after weight decay.
        """
        if self.weight_decay > 0:
-            params = self.params
+            params = self.parameters
            gradients = self.hyper_map(F.partial(apply_decay, self.weight_decay), self.decay_flags, params, gradients)

        return gradients