Skip to content

GitLab

Commit c9a6456d authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Fix a bug when initializing the model with all modules. 

PiperOrigin-RevId: 459611611
parent baea8882
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/tasks/video_classification.py
View file @ c9a6456d
...@@ -85,7 +85,7 @@ class VideoClassificationTask(base_task.Task): ...@@ -85,7 +85,7 @@ class VideoClassificationTask(base_task.Task):
# Restoring checkpoint. # Restoring checkpoint.
if self.task_config.init_checkpoint_modules == 'all': 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 = ckpt.read(ckpt_dir_or_file)
status.expect_partial().assert_existing_objects_matched() status.expect_partial().assert_existing_objects_matched()
elif self.task_config.init_checkpoint_modules == 'backbone': elif self.task_config.init_checkpoint_modules == 'backbone':
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment