Fix no such file or directory error (huggingface#23783)

* Fix no such file or directory error * Address comment * Fix formatting issue
sheonhan · Jun 1, 2023 · 6c47f56 · 6c47f56
1 parent 6673629
commit 6c47f56
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/tensorflow/image-classification/run_image_classification.py b/examples/tensorflow/image-classification/run_image_classification.py
@@ -543,6 +543,7 @@ def compute_metrics(p):
                 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))