Internal change

PiperOrigin-RevId: 520504242

Author: arashwan

official/vision/tasks/maskrcnn.py

@@ -105,7 +105,7 @@ def initialize(self, model: tf.keras.Model):
 
     # Restoring checkpoint.
     if self.task_config.init_checkpoint_modules == 'all':
-      ckpt = tf.train.Checkpoint(**model.checkpoint_items)
+      ckpt = tf.train.Checkpoint(model=model)
       status = ckpt.read(ckpt_dir_or_file)
       status.expect_partial().assert_existing_objects_matched()
     else: