Internal change

PiperOrigin-RevId: 346397839

Internal change
PiperOrigin-RevId: 346397839
77d47f79 · Chen Chen · A. Unique TensorFlower · c5f11e3f · 77d47f79
Commit 77d47f79 authored Dec 08, 2020 by Chen Chen Committed by A. Unique TensorFlower Dec 08, 2020
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.rmtree(file_to_remove)
+      tf.io.gfile.remove(file_to_remove)
    checkpoint.save(self.best_ckpt_path)
  @property