Add metrics plots

roboflow · Aug 29, 2024 · 783dc88 · 783dc88
1 parent 7c62cae
commit 783dc88
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/maestro/trainer/models/florence_2/metrics.py b/maestro/trainer/models/florence_2/metrics.py
@@ -200,6 +200,7 @@ def summarise_metrics(
     split_name: str,
 ) -> None:
     plots_dir_path = os.path.join(training_dir, "metrics", split_name)
+    os.makedirs(plots_dir_path, exist_ok=True)
     for metric_name in metrics_tracker.describe_metrics():
         plot_path = os.path.join(plots_dir_path, f"metric_{metric_name}_plot.png")
         plt.clf()