Browse Source

!6459 remove comment

Merge pull request !6459 from lijiaqi/remove_comment
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
e4d8865a6b
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      mindspore/nn/wrap/loss_scale.py

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

@@ -174,8 +174,7 @@ class TrainOneStepWithLossScaleCell(TrainOneStepCell):
TrainOneStepWithLossScaleCell will be compiled to be graph which takes `*inputs` as input data. 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 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 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: Args:
network (Cell): The training network. The network only supports single output. network (Cell): The training network. The network only supports single output.


Loading…
Cancel
Save