Browse Source

!6430 fix sparse loss scale

Merge pull request !6430 from riemann_penn/fix_sparse_loss_scale
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
cd88f58b88
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/nn/wrap/loss_scale.py

+ 3
- 0
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"""


Loading…
Cancel
Save