SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix savedir for by epoch (#16996)

Browse Source
This commit is contained in:
Zachary Mueller
2022-04-28 13:49:45 -04:00
committed by GitHub
parent 5af5735f62
commit 3486a92a57
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/translation/run_translation_no_trainer.py
View File

@@ -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)