p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!3415 fix bug of group lr 

Merge pull request !3415 from wangnan39/fix_bug_of_tuple_lr_in_optimizer
This commit is contained in:
mindspore-ci-bot 2020-07-24 17:05:34 +08:00 committed by Gitee
parent 2f1a5b979d 4c40fd681e
commit 20cb5eec66
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/nn/optim/optimizer.py
View File

@ -130,7 +130,7 @@ class Optimizer(Cell):
if self.dynamic_lr: if self.dynamic_lr:
self.learning_rate = CellList(self.group_lr) self.learning_rate = CellList(self.group_lr)
else: else:
self.learning_rate = tuple(self.group_lr) self.learning_rate = ParameterTuple(self.group_lr)
else: else:
self.learning_rate = self._build_single_lr(learning_rate, 'learning_rate') self.learning_rate = self._build_single_lr(learning_rate, 'learning_rate')
if self.is_group: if self.is_group: