Browse Source

!1043 FIxed a simple question

Merge pull request !1043 from zhangzheng/simple
tags/v0.3.0-alpha
mindspore-ci-bot Gitee 6 years ago
parent
commit
3c4c0da8ae
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/wrap/loss_scale.py

+ 1
- 1
mindspore/nn/wrap/loss_scale.py View File

@@ -187,7 +187,7 @@ class TrainOneStepWithLossScaleCell(Cell):
Examples:
>>> net_with_loss = Net()
>>> optimizer = nn.Momentum(net_with_loss.trainable_params(), learning_rate=0.1, momentum=0.9)
>>> manager = nn.DynamicLossScaleUpdateCell(init_loss_scale=2**12, scale_factor=2, scale_window=1000)
>>> manager = nn.DynamicLossScaleUpdateCell(loss_scale_value=2**12, scale_factor=2, scale_window=1000)
>>> train_network = nn.TrainOneStepWithLossScaleCell(net_with_loss, optimizer, scale_update_cell=manager)
>>> train_network.set_train()
>>>


Loading…
Cancel
Save