From 3486a92a5725c6b35798b64d8ce973aa28d78252 Mon Sep 17 00:00:00 2001 From: Zachary Mueller Date: Thu, 28 Apr 2022 13:49:45 -0400 Subject: [PATCH] Fix savedir for by epoch (#16996) --- examples/pytorch/translation/run_translation_no_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/pytorch/translation/run_translation_no_trainer.py b/examples/pytorch/translation/run_translation_no_trainer.py index 597c5c353e..f01267a288 100644 --- a/examples/pytorch/translation/run_translation_no_trainer.py +++ b/examples/pytorch/translation/run_translation_no_trainer.py @@ -678,7 +678,7 @@ def main(): ) if args.checkpointing_steps == "epoch": - output_dir = f"step_{completed_steps}" + output_dir = f"epoch_{epoch}" if args.output_dir is not None: output_dir = os.path.join(args.output_dir, output_dir) accelerator.save_state(output_dir)