Internal change

arashwan · tensorflower-gardener · commit 209a259d2161 · 2023-03-29T19:07:42.000-07:00
PiperOrigin-RevId: 520504242
diff --git a/official/vision/tasks/maskrcnn.py b/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:
