diff --git a/ppdet/modeling/assigners/task_aligned_assigner_cr.py b/ppdet/modeling/assigners/task_aligned_assigner_cr.py
index 5c5097604d55856af45e0eeffacd381c0f5518b0..4558d6e8ec7af5a59fc4975bff089616f0b0b209 100644
--- a/ppdet/modeling/assigners/task_aligned_assigner_cr.py
+++ b/ppdet/modeling/assigners/task_aligned_assigner_cr.py
@@ -178,4 +178,4 @@ class TaskAlignedAssigner_CR(nn.Layer):
         alignment_metrics = alignment_metrics.max(-2).unsqueeze(-1)
         assigned_scores = assigned_scores * alignment_metrics
 
-        return assigned_labels, assigned_bboxes, assigned_scores
+        return assigned_labels, assigned_bboxes, assigned_scores, mask_positive