Internal change

PiperOrigin-RevId: 324700979

Internal change
PiperOrigin-RevId: 324700979
7ebdee5f · Chen Chen · A. Unique TensorFlower · 510736ba · 7ebdee5f
Commit 7ebdee5f authored Aug 03, 2020 by Chen Chen Committed by A. Unique TensorFlower Aug 03, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/core/base_task.py official/core/base_task.py +1 -1

No files found.
--- a/official/core/base_task.py
+++ b/official/core/base_task.py
@@ -76,7 +76,7 @@ class Task(tf.Module):
      return

    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()
    logging.info("Finished loading pretrained checkpoint from %s",
                 ckpt_dir_or_file)