Browse Source

!3415 fix bug of group lr

Merge pull request !3415 from wangnan39/fix_bug_of_tuple_lr_in_optimizer
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
20cb5eec66
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/optim/optimizer.py

+ 1
- 1
mindspore/nn/optim/optimizer.py View File

@@ -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:


Loading…
Cancel
Save