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

Fix result saving errors of pytorch examples (#20276)

Browse Source
This commit is contained in:
Jiahao Li
2022-11-16 22:51:04 +08:00
committed by GitHub
parent e627e9b5ae
commit 9681f052a1
8 changed files with 26 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
examples/pytorch/summarization/run_summarization_no_trainer.py
View File

@@ -747,16 +747,10 @@ def main():
tokenizer.save_pretrained(args.output_dir)
if args.push_to_hub:
repo.push_to_hub(commit_message="End of training", auto_lfs_prune=True)
with open(os.path.join(args.output_dir, "all_results.json"), "w") as f:
json.dump(
{
"eval_rouge1": result["rouge1"],
"eval_rouge2": result["rouge2"],
"eval_rougeL": result["rougeL"],
"eval_rougeLsum": result["rougeLsum"],
},
f,
)
all_results = {f"eval_{k}": v for k, v in result.items()}
with open(os.path.join(args.output_dir, "all_results.json"), "w") as f:
json.dump(all_results, f)
if __name__ == "__main__":