Skip to content

GitLab

Unverified Commit 3486a92a authored by Zachary Mueller's avatar Zachary Mueller Committed by GitHub
Browse files

Fix savedir for by epoch (#16996)

parent 5af5735f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/translation/run_translation_no_trainer.py
View file @ 3486a92a
...@@ -678,7 +678,7 @@ def main(): ...@@ -678,7 +678,7 @@ def main():
) )
if args.checkpointing_steps == "epoch": if args.checkpointing_steps == "epoch":
output_dir = f"step_{completed_steps}" output_dir = f"epoch_{epoch}"
if args.output_dir is not None: if args.output_dir is not None:
output_dir = os.path.join(args.output_dir, output_dir) output_dir = os.path.join(args.output_dir, output_dir)
accelerator.save_state(output_dir) accelerator.save_state(output_dir)
......
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