From 2f4edeffff82e7e2c21c719503f9b5865aa9e0f4 Mon Sep 17 00:00:00 2001 From: mbsantiago Date: Sun, 31 Aug 2025 19:22:45 +0100 Subject: [PATCH] Create separate preprocessor for the train/val datasets --- src/batdetect2/train/train.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/batdetect2/train/train.py b/src/batdetect2/train/train.py index 32d43c4..0178333 100644 --- a/src/batdetect2/train/train.py +++ b/src/batdetect2/train/train.py @@ -16,6 +16,7 @@ from batdetect2.evaluate.metrics import ( ) from batdetect2.models import Model, build_model from batdetect2.plotting.clips import AudioLoader, build_audio_loader +from batdetect2.preprocess import build_preprocessor from batdetect2.train.augmentations import ( RandomAudioSource, build_augmentations, @@ -75,7 +76,7 @@ def train( train_annotations, audio_loader=audio_loader, labeller=labeller, - preprocessor=model.preprocessor, + preprocessor=build_preprocessor(config.preprocess), config=config.train, num_workers=train_workers, ) @@ -85,7 +86,7 @@ def train( val_annotations, audio_loader=audio_loader, labeller=labeller, - preprocessor=model.preprocessor, + preprocessor=build_preprocessor(config.preprocess), config=config.train, num_workers=val_workers, )