fix_bug_of_group_lr_when_save_ckpt

6 years ago · 3c93ff3385
parent f9aec99c01
commit 3c93ff3385
1 changed files with 1 additions and 1 deletions
--- a/mindspore/nn/optim/optimizer.py
+++ b/mindspore/nn/optim/optimizer.py
@ -130,7 +130,7 @@ class Optimizer(Cell):
            if self.dynamic_lr:
                self.learning_rate = CellList(self.group_lr)
            else:
-                self.learning_rate = tuple(self.group_lr)
+                self.learning_rate = ParameterTuple(self.group_lr)
        else:
            self.learning_rate = self._build_single_lr(learning_rate, 'learning_rate')
        if self.is_group: