smoothL1Loss grad wrt target

5 years ago · a5c021f450
parent 10076ffe1a
commit a5c021f450
1 changed files with 2 additions and 1 deletions
--- a/mindspore/ops/_grad/grad_nn_ops.py
+++ b/mindspore/ops/_grad/grad_nn_ops.py
@ -502,7 +502,8 @@ def get_bprop_smooth_l1_loss(self):

    def bprop(prediction, target, out, dout):
        dx = grad(prediction, target, dout)
-        return dx, zeros_like(target)
+        dy = grad(target, prediction, dout)
+        return dx, dy

    return bprop