Fix no such file or directory error (#23783)

* Fix no such file or directory error

* Address comment

* Fix formatting issue
This commit is contained in:
Ran Ran 2023-05-26 11:24:57 -07:00 committed by GitHub
parent b7b729b38d
commit e724246935
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -543,6 +543,7 @@ def main():
logging.info(f"{metric_name}: {value:.3f}")
if training_args.output_dir is not None:
os.makedirs(training_args.output_dir, exist_ok=True)
with open(os.path.join(training_args.output_dir, "all_results.json"), "w") as f:
f.write(json.dumps(eval_metrics))