From b27e0740ba98f689340bf38fa053d9f72967400f Mon Sep 17 00:00:00 2001 From: zhangz0911gm Date: Tue, 14 Apr 2020 21:58:04 -0400 Subject: [PATCH] Getting a simple issues fixed --- mindspore/nn/wrap/loss_scale.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/nn/wrap/loss_scale.py b/mindspore/nn/wrap/loss_scale.py index 65d66f0150..a2f39b7b05 100644 --- a/mindspore/nn/wrap/loss_scale.py +++ b/mindspore/nn/wrap/loss_scale.py @@ -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() >>>