提交 a5c021f4 编写于 作者: Z zhaozhenlong

smoothL1Loss grad wrt target

上级 10076ffe
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册