From 4c40fd681e7d6245ed9c1fdb1012102d5c86ab68 Mon Sep 17 00:00:00 2001 From: "wangnan39@huawei.com" Date: Fri, 24 Jul 2020 13:30:38 +0800 Subject: [PATCH] fix bug of group lr --- mindspore/nn/optim/optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/nn/optim/optimizer.py b/mindspore/nn/optim/optimizer.py index 9379e395ae..281bd045c8 100755 --- 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: