Internal change

PiperOrigin-RevId: 354647274

Internal change
PiperOrigin-RevId: 354647274
30050e23 · A. Unique TensorFlower · d0060ec4 · 30050e23
Commit 30050e23 authored Jan 29, 2021 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/official/core/train_utils.py
+++ b/official/core/train_utils.py
@@ -108,7 +108,7 @@ class BestCheckpointExporter:
    # Saving the best checkpoint might be interrupted if the job got killed.
    for file_to_remove in tf.io.gfile.glob(self.best_ckpt_path + '*'):
      tf.io.gfile.remove(file_to_remove)
-    checkpoint.save(self.best_ckpt_path)
+    checkpoint.write(self.best_ckpt_path)

  @property
  def best_ckpt_logs(self):