Save checkpoint on max class mAP

2026-01-09 16:59:33 +01:00 · 2025-09-30 18:20:31 +01:00 · 2025-09-30 18:20:31 +01:00 · 8727edf466
commit 8727edf466
parent 2f48c58de1
1 changed files with 1 additions and 0 deletions
--- 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),
    ]