riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats 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":
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)