Fix a bug when initializing the model with all modules.

PiperOrigin-RevId: 459611611

Fix a bug when initializing the model with all modules.
PiperOrigin-RevId: 459611611
cfdc930e · A. Unique TensorFlower · 331dcdf1 · cfdc930e
Commit cfdc930e authored Jul 07, 2022 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/tasks/video_classification.py official/vision/tasks/video_classification.py +1 -1

No files found.
--- a/official/vision/tasks/video_classification.py
+++ b/official/vision/tasks/video_classification.py
@@ -85,7 +85,7 @@ class VideoClassificationTask(base_task.Task):
    # 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':