From 7d416e0f991e69f0f4648bdcc9cd569581e9b0bd Mon Sep 17 00:00:00 2001 From: mbsantiago Date: Tue, 5 May 2026 00:55:27 +0100 Subject: [PATCH] Set model mode during training --- src/batdetect2/api_v2.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/batdetect2/api_v2.py b/src/batdetect2/api_v2.py index 9f872b5..0513d63 100644 --- a/src/batdetect2/api_v2.py +++ b/src/batdetect2/api_v2.py @@ -109,6 +109,7 @@ class BatDetect2API: ): from batdetect2.train import run_train + self.model.train() run_train( train_annotations=train_annotations, val_annotations=val_annotations, @@ -130,6 +131,7 @@ class BatDetect2API: audio_config=audio_config or self.audio_config, logger_config=logger_config or self.logging_config.train, ) + self.model.eval() return self def finetune(