Expect partial checkpoints by default
@@ -141,7 +141,6 @@ class Checkpoint(object):
checkpoint_path=checkpoint_path, weights_only=weights_only
)
load_status = checkpoint.restore(checkpoint_path)
- if weights_only:
load_status.expect_partial()
tf.get_logger().info("Restored checkpoint %s", checkpoint_path)
return checkpoint_path
