Fix no such file or directory error (#23783)

* Fix no such file or directory error * Address comment * Fix formatting issue

Fix no such file or directory error (#23783)
* Fix no such file or directory error * Address comment * Fix formatting issue
e7242469 · Ran Ran · GitHub · b7b729b3 · e7242469
Unverified Commit e7242469 authored May 26, 2023 by Ran Ran Committed by GitHub May 26, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

examples/tensorflow/image-classification/run_image_classification.py ...nsorflow/image-classification/run_image_classification.py +1 -0

No files found.
--- a/examples/tensorflow/image-classification/run_image_classification.py
+++ b/examples/tensorflow/image-classification/run_image_classification.py
@@ -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))