Internal change

PiperOrigin-RevId: 397851196

Author: tensorflower-gardener

official/vision/beta/tasks/image_classification.py

@@ -62,7 +62,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()
     elif self.task_config.init_checkpoint_modules == 'backbone':