mirror of
https://github.com/macaodha/batdetect2.git
synced 2026-04-04 15:20:19 +02:00
Add save last config to checkpoint
This commit is contained in:
parent
32d8c4a9e5
commit
875751d340
@ -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,
|
||||
)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user