提交 20cb5eec 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!3415 fix bug of group lr

Merge pull request !3415 from wangnan39/fix_bug_of_tuple_lr_in_optimizer
......@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册