m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!1043 FIxed a simple question

Browse Source 
Merge pull request !1043 from zhangzheng/simple
pull/1043/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
parent 14fa64cddb b27e0740ba
commit 3c4c0da8ae
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
mindspore/nn/wrap/loss_scale.py
Unescape View File

@ -187,7 +187,7 @@ class TrainOneStepWithLossScaleCell(Cell):
Examples: Examples:
>>> net_with_loss = Net() >>> net_with_loss = Net()
>>> optimizer = nn.Momentum(net_with_loss.trainable_params(), learning_rate=0.1, momentum=0.9) >>> 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 = nn.TrainOneStepWithLossScaleCell(net_with_loss, optimizer, scale_update_cell=manager)
>>> train_network.set_train() >>> train_network.set_train()
>>> >>>

Write Preview
Loading…
Cancel
Save