From cca1d82d63e64103a1a712e226d7549526a9d589 Mon Sep 17 00:00:00 2001 From: mbsantiago Date: Sun, 31 Aug 2025 19:10:06 +0100 Subject: [PATCH] Remove blocking .to(device) --- src/batdetect2/preprocess/spectrogram.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/batdetect2/preprocess/spectrogram.py b/src/batdetect2/preprocess/spectrogram.py index b79ef81..067a965 100644 --- a/src/batdetect2/preprocess/spectrogram.py +++ b/src/batdetect2/preprocess/spectrogram.py @@ -370,10 +370,10 @@ class SpectrogramPipeline(torch.nn.Module): self.resizer = resizer def forward(self, wav: torch.Tensor) -> torch.Tensor: - spec = self.spec_builder.to(wav)(wav) - spec = self.freq_cutter.to(wav)(spec) - spec = self.transforms.to(wav)(spec) - return self.resizer.to(wav)(spec) + spec = self.spec_builder(wav) + spec = self.freq_cutter(spec) + spec = self.transforms(spec) + return self.resizer(spec) def compute_spectrogram(self, wav: torch.Tensor) -> torch.Tensor: return self.spec_builder(wav)