lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-05-22 22:32:18 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Modify max epochs with override before logging

Browse Source
This commit is contained in:
mbsantiago 2026-05-04 17:12:55 +01:00
parent 44f9870e9e
commit ae4f742345
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -215,18 +215,15 @@ def build_trainer(
run_name=run_name,
)
if num_epochs is not None:
trainer_conf.max_epochs = num_epochs
train_logger.log_hyperparams(
config.model_dump(
mode="json",
exclude_none=True,
)
config.model_dump(mode="json", exclude_none=True)
)
train_config = trainer_conf.model_dump(exclude_none=True)
if num_epochs is not None:
train_config["max_epochs"] = num_epochs
return Trainer(
**train_config,
logger=train_logger,