Fix savedir for by epoch (#16996)

3486a92a · Zachary Mueller · GitHub · 5af5735f · 3486a92a
Unverified Commit 3486a92a authored Apr 28, 2022 by Zachary Mueller Committed by GitHub Apr 28, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/pytorch/translation/run_translation_no_trainer.py examples/pytorch/translation/run_translation_no_trainer.py +1 -1

No files found.
--- 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)