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

9
examples/pytorch/token-classification/run_ner_no_trainer.py
View File

@@ -766,10 +766,11 @@ def main():
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_accuracy": eval_metric["accuracy"], "train_loss": total_loss.item() / len(train_dataloader)}, f
)
all_results = {f"eval_{k}": v for k, v in eval_metric.items()}
if args.with_tracking:
all_results.update({"train_loss": total_loss.item() / len(train_dataloader)})
with open(os.path.join(args.output_dir, "all_results.json"), "w") as f:
json.dump(all_results, f)
if __name__ == "__main__":