Silence warnings about unmatched checkpoint objects.

PiperOrigin-RevId: 378154667

Silence warnings about unmatched checkpoint objects.
PiperOrigin-RevId: 378154667
acdb71d6 · Vighnesh Birodkar · TF Object Detection Team · b09e7582 · acdb71d6
Commit acdb71d6 authored Jun 08, 2021 by Vighnesh Birodkar Committed by TF Object Detection Team Jun 08, 2021
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

research/object_detection/model_lib_v2.py research/object_detection/model_lib_v2.py +2 -1

No files found.
--- a/research/object_detection/model_lib_v2.py
+++ b/research/object_detection/model_lib_v2.py
@@ -397,7 +397,8 @@ def load_fine_tune_checkpoint(model, checkpoint_path, checkpoint_type,
      fine_tune_checkpoint_type=checkpoint_type)
  validate_tf_v2_checkpoint_restore_map(restore_from_objects_dict)
  ckpt = tf.train.Checkpoint(**restore_from_objects_dict)
-  ckpt.restore(checkpoint_path).assert_existing_objects_matched()
+  ckpt.restore(
+      checkpoint_path).expect_partial().assert_existing_objects_matched()


 def get_filepath(strategy, filepath):