@@ -1406,6 +1406,7 @@ class RoIHeads(nn.Module):
loss_ins = compute_ins_loss(feature_logits, ins_proposals, gt_inses,ins_pos_matched_idxs)
total_loss, loss_arc_equation, loss_arc_ends = compute_arc_equation_loss(arc_equation,
ins_proposals,
+
gt_mask_ends,
gt_mask_params,
ins_pos_matched_idxs,