p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!6430 fix sparse loss scale 

Merge pull request !6430 from riemann_penn/fix_sparse_loss_scale
This commit is contained in:
mindspore-ci-bot 2020-09-18 09:02:01 +08:00 committed by Gitee
parent 1256737a7c 9d8b14d942
commit cd88f58b88
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/nn/wrap/loss_scale.py
View File

@ -48,6 +48,9 @@ grad_overflow = P.FloatStatus()
def _tensor_grad_overflow(grad):
return grad_overflow(grad)
@_grad_overflow.register("RowTensor")
def _tensor_grad_overflow_row_tensor(grad):
return grad_overflow(grad.values)
class DynamicLossScaleUpdateCell(Cell):
r"""