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

Set model mode during training

Browse Source
This commit is contained in:
mbsantiago 2026-05-05 00:55:27 +01:00
parent 2d0b810ed3
commit 7d416e0f99
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/api_v2.py
View File

@ -109,6 +109,7 @@ class BatDetect2API:
): ):
from batdetect2.train import run_train from batdetect2.train import run_train
self.model.train()
run_train( run_train(
train_annotations=train_annotations, train_annotations=train_annotations,
val_annotations=val_annotations, val_annotations=val_annotations,
@ -130,6 +131,7 @@ class BatDetect2API:
audio_config=audio_config or self.audio_config, audio_config=audio_config or self.audio_config,
logger_config=logger_config or self.logging_config.train, logger_config=logger_config or self.logging_config.train,
) )
self.model.eval()
return self return self
def finetune( def finetune(