Fix result saving errors of pytorch examples (#20276)

2022-11-16 22:51:04 +08:00
parent e627e9b5ae
commit 9681f052a1
8 changed files with 26 additions and 27 deletions
--- a/examples/pytorch/text-classification/run_glue_no_trainer.py
+++ b/examples/pytorch/text-classification/run_glue_no_trainer.py
@@ -625,8 +625,9 @@ def main():
        logger.info(f"mnli-mm: {eval_metric}")

    if args.output_dir is not None:
+        all_results = {f"eval_{k}": v for k, v in eval_metric.items()}
        with open(os.path.join(args.output_dir, "all_results.json"), "w") as f:
-            json.dump({"eval_accuracy": eval_metric["accuracy"]}, f)
+            json.dump(all_results, f)


 if __name__ == "__main__":