diff --git a/ppdet/modeling/losses/iou_loss.py b/ppdet/modeling/losses/iou_loss.py index 56532c868d537c688867e78a7a15b270639f2ba1..eb0e7112b4181646991367bd8d564d202566cf66 100644 --- a/ppdet/modeling/losses/iou_loss.py +++ b/ppdet/modeling/losses/iou_loss.py @@ -176,9 +176,9 @@ class IouLoss(object): grid_y_act.stop_gradient = True if is_gt: cx = fluid.layers.elementwise_add(dcx, gi) / grid_x_act - cx.gradient = True + cx.stop_gradient = True cy = fluid.layers.elementwise_add(dcy, gj) / grid_y_act - cy.gradient = True + cy.stop_gradient = True else: dcx_sig = fluid.layers.sigmoid(dcx) dcy_sig = fluid.layers.sigmoid(dcy)