lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-01-10 17:19:34 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Better order for checkpoints

Browse Source
This commit is contained in:
mbsantiago 2025-09-09 15:56:46 +01:00
parent 16a0fa7b75
commit 41b18c3f0a
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/batdetect2/train/train.py
View File

@ -152,6 +152,12 @@ def build_trainer_callbacks(
if checkpoint_dir is None: if checkpoint_dir is None:
checkpoint_dir = DEFAULT_CHECKPOINT_DIR checkpoint_dir = DEFAULT_CHECKPOINT_DIR
if experiment_name is not None:
checkpoint_dir = checkpoint_dir / experiment_name
if run_name is not None:
checkpoint_dir = checkpoint_dir / run_name
filename = "best-{epoch:02d}-{val_loss:.0f}" filename = "best-{epoch:02d}-{val_loss:.0f}"
if run_name is not None: if run_name is not None: