diff --git a/src/batdetect2/train/train.py b/src/batdetect2/train/train.py index 1f1745a..8845fa8 100644 --- a/src/batdetect2/train/train.py +++ b/src/batdetect2/train/train.py @@ -73,9 +73,8 @@ def run_train( audio_loader = audio_loader or build_audio_loader(config=audio_config) - if model is not None: - preprocessor = preprocessor or model.preprocessor - + # NOTE: Create a new preprocessor instead of using the one from the model + # to avoid issues with device placement. preprocessor = preprocessor or build_preprocessor( input_samplerate=audio_loader.samplerate, config=model_config.preprocess,