pgjmblvch
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!6459 remove comment 

Merge pull request !6459 from lijiaqi/remove_comment
This commit is contained in:
mindspore-ci-bot 2020-09-18 14:21:22 +08:00 committed by Gitee
parent 1dc71651ae c5c1470bf1
commit e4d8865a6b
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
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.
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.