qsdy
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix_bug_of_group_lr_when_save_ckpt

This commit is contained in:
wangnan39@huawei.com 2020-07-24 13:40:14 +08:00
parent f9aec99c01
commit 3c93ff3385
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:
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: