Golden path

bf34a252 · jinoobaek-qz · Lysandre Debut · 528d3f32 · bf34a252
Commit bf34a252 authored Oct 07, 2019 by jinoobaek-qz Committed by Lysandre Debut Oct 09, 2019
Show whitespace changes
Inline Side-by-side

Showing with 19 additions and 17 deletions

examples/run_lm_finetuning.py examples/run_lm_finetuning.py +19 -17

No files found.
--- a/examples/run_lm_finetuning.py
+++ b/examples/run_lm_finetuning.py
@@ -114,7 +114,9 @@ def _rotate_checkpoints(args, checkpoint_prefix, use_mtime=False):

    # Check if we should delete older checkpoint(s)
    glob_checkpoints = glob.glob(os.path.join(args.output_dir, '{}-*'.format(checkpoint_prefix)))
-    if len(glob_checkpoints) > args.save_total_limit:
+    if len(glob_checkpoints) <= args.save_total_limit:
+        return
+
    checkpoints_sorted = []
    for path in glob_checkpoints:
        regex_match = re.match('.*{}-([0-9]+)'.format(checkpoint_prefix), path)