From 875751d3400a706f28f18210c8700896aa27f110 Mon Sep 17 00:00:00 2001
From: Santiago Martinez Balvanera <santiago.mbal@gmail.com>
Date: Thu, 19 Mar 2026 00:36:45 +0000
Subject: [PATCH] Add save last config to checkpoint

---
 src/batdetect2/train/checkpoints.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/batdetect2/train/checkpoints.py b/src/batdetect2/train/checkpoints.py
index fb68503..92db951 100644
--- a/src/batdetect2/train/checkpoints.py
+++ b/src/batdetect2/train/checkpoints.py
@@ -18,6 +18,7 @@ class CheckpointConfig(BaseConfig):
     mode: str = "max"
     save_top_k: int = 1
     filename: str | None = None
+    save_last: bool = False
 
 
 def build_checkpoint_callback(
@@ -43,4 +44,5 @@ def build_checkpoint_callback(
         monitor=config.monitor,
         mode=config.mode,
         filename=config.filename,
+        save_last=config.save_last,
     )