lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2025-06-29 14:41:58 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix trainer init

Browse Source
This commit is contained in:
mbsantiago 2025-06-26 19:40:27 -06:00
parent ce15a0f152
commit b396d4908a
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -101,8 +101,7 @@ def build_trainer(
targets: TargetProtocol, targets: TargetProtocol,
) -> Trainer: ) -> Trainer:
trainer_conf = PLTrainerConfig.model_validate( trainer_conf = PLTrainerConfig.model_validate(
conf.train, conf.train.model_dump(mode="python")
from_attributes=True,
) )
return Trainer( return Trainer(
**trainer_conf.model_dump(exclude_none=True), **trainer_conf.model_dump(exclude_none=True),