diff --git a/ppdet/modeling/architecture/cascade_rcnn.py b/ppdet/modeling/architecture/cascade_rcnn.py
index 1a678b6a076d12a3607199aab196d9b013b2eb9b..1cfdaea5122bfbdec30891d827f1f7a7618fe50f 100644
--- a/ppdet/modeling/architecture/cascade_rcnn.py
+++ b/ppdet/modeling/architecture/cascade_rcnn.py
@@ -165,5 +165,5 @@ class CascadeRCNN(BaseArch):
             'bbox_num': bbox_num,
         }
         if self.with_mask:
-            output.update(self.mask_head_out)
+            output.update({'mask': self.mask_head_out})
         return output