lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-04-04 15:20:19 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add save last config to checkpoint

Browse Source
This commit is contained in:
Santiago Martinez Balvanera 2026-03-19 00:36:45 +00:00
parent 32d8c4a9e5
commit 875751d340
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/batdetect2/train/checkpoints.py
View File

@ -18,6 +18,7 @@ class CheckpointConfig(BaseConfig):
mode: str = "max" mode: str = "max"
save_top_k: int = 1 save_top_k: int = 1
filename: str | None = None filename: str | None = None
save_last: bool = False
def build_checkpoint_callback( def build_checkpoint_callback(
@ -43,4 +44,5 @@ def build_checkpoint_callback(
monitor=config.monitor, monitor=config.monitor,
mode=config.mode, mode=config.mode,
filename=config.filename, filename=config.filename,
save_last=config.save_last,
) )