Skip to content

GitLab

Commit 8f34962f authored by Chen Chen's avatar Chen Chen Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 324700979
parent 6b692770
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/core/base_task.py
View file @ 8f34962f
...@@ -76,7 +76,7 @@ class Task(tf.Module): ...@@ -76,7 +76,7 @@ class Task(tf.Module):
return return
ckpt = tf.train.Checkpoint(**model.checkpoint_items) ckpt = tf.train.Checkpoint(**model.checkpoint_items)
status = ckpt.restore(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()
logging.info("Finished loading pretrained checkpoint from %s", logging.info("Finished loading pretrained checkpoint from %s",
ckpt_dir_or_file) ckpt_dir_or_file)
......
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