Browse Source

fix bug in example of Lamb

tags/v1.1.0
wangnan39@huawei.com 5 years ago
parent
commit
9b545e4982
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/optim/lamb.py

+ 1
- 1
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()


Loading…
Cancel
Save