From 8727edf466156e5ef42b6c656d931da655eaceb1 Mon Sep 17 00:00:00 2001
From: mbsantiago <santiago.mbal@gmail.com>
Date: Tue, 30 Sep 2025 18:20:31 +0100
Subject: [PATCH] Save checkpoint on max class mAP

---
 src/batdetect2/train/train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/batdetect2/train/train.py b/src/batdetect2/train/train.py
index aed25cd..b0f1642 100644
--- a/src/batdetect2/train/train.py
+++ b/src/batdetect2/train/train.py
@@ -147,6 +147,7 @@ def build_trainer_callbacks(
             dirpath=str(checkpoint_dir),
             save_top_k=1,
             monitor="classification/mean_average_precision",
+            mode="max",
         ),
         ValidationMetrics(evaluator),
     ]