Add metrics plots

roboflow · Aug 29, 2024 · dbe7593 · dbe7593
1 parent 783dc88
commit dbe7593
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/maestro/trainer/models/florence_2/metrics.py b/maestro/trainer/models/florence_2/metrics.py
@@ -138,7 +138,12 @@ def dump_post_processed_outputs(
     result_dict = {
         dataset.dataset.entries[idx]["image"]: output for idx, output in enumerate(post_processed_text_outputs)
     }
-    target_path = os.path.join(training_dir, split_name, f"post_processed_text_predictions_{split_name}.json")
+    target_path = os.path.join(
+        training_dir,
+        "model_debug",
+        split_name,
+        f"post_processed_text_predictions_{split_name}.json",
+    )
     save_json(path=target_path, content=result_dict)