Skip to content

GitLab

"examples/pytorch/vscode:/vscode.git/clone" did not exist on "1e69bf538774894eadfe21b76f047a515bdde322"
Commit 77d47f79 authored by Chen Chen's avatar Chen Chen Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 346397839
parent c5f11e3f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/core/train_utils.py
View file @ 77d47f79
...@@ -108,7 +108,7 @@ class BestCheckpointExporter: ...@@ -108,7 +108,7 @@ class BestCheckpointExporter:
# Saving the best checkpoint might be interrupted if the job got killed. # 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 + '*'): 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) checkpoint.save(self.best_ckpt_path)
@property @property
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment