lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-01-10 17:19:34 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Remove blocking .to(device)

Browse Source
This commit is contained in:
mbsantiago 2025-08-31 19:10:06 +01:00
parent 55f473c9ca
commit cca1d82d63
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/batdetect2/preprocess/spectrogram.py
View File

@ -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)