diff --git a/mindspore/nn/wrap/loss_scale.py b/mindspore/nn/wrap/loss_scale.py index 729119db9c..47a66d7ffd 100644 --- a/mindspore/nn/wrap/loss_scale.py +++ b/mindspore/nn/wrap/loss_scale.py @@ -174,8 +174,7 @@ class TrainOneStepWithLossScaleCell(TrainOneStepCell): TrainOneStepWithLossScaleCell will be compiled to be graph which takes `*inputs` as input data. The Tensor type of `scale_sense` is acting as loss scaling value. If you want to update it on host side, the value should be provided. If the Tensor type of `scale_sense` is not given, the loss scale update logic - should be provied by Cell type of `scale_sense`. If Cell type of `scale_sense` is not None and Tensor type - of `scale_sense` is provided, the Cell type of `scale_sense` will be ignored. + should be provied by Cell type of `scale_sense`. Args: network (Cell): The training network. The network only supports single output.