No public description

PiperOrigin-RevId: 590906048

Author: tensorflower-gardener

official/vision/modeling/maskrcnn_model.py

@@ -495,7 +495,7 @@ def checkpoint_items(
         detection_head=self.detection_head)
     if self.decoder is not None:
       items.update(decoder=self.decoder)
-    if self._include_mask:
+    if self._include_mask and self.mask_head is not None:
       items.update(mask_head=self.mask_head)
 
     return items