From dbe7593f9e644d5f203d8421e5297ed6dc768d31 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pawe=C5=82=20P=C4=99czek?= Date: Thu, 29 Aug 2024 17:31:23 +0200 Subject: [PATCH] Add metrics plots --- maestro/trainer/models/florence_2/metrics.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/maestro/trainer/models/florence_2/metrics.py b/maestro/trainer/models/florence_2/metrics.py index d45c0b0..5746665 100644 --- 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)