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

!9273 fix bug in example of Lamb 

From: @wangnan39
Reviewed-by: @zh_qh,@kingxian
Signed-off-by: @kingxian
This commit is contained in:
mindspore-ci-bot 2020-12-01 21:27:14 +08:00 committed by Gitee
parent c5448cfa23 9b545e4982
commit 9cf4d0d0dc
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/lamb.py
View File

@ -241,7 +241,7 @@ class Lamb(Optimizer):
Examples:
>>> net = Net()
>>> #1) All parameters use the same learning rate and weight decay
>>> optim = nn.Lamb(params=net.trainable_params(learning_rate=0.1))
>>> optim = nn.Lamb(params=net.trainable_params(), learning_rate=0.1)
>>>
>>> #2) Use parameter groups and set different values
>>> poly_decay_lr = learning_rate_schedule.PolynomialDecayLR()