lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2025-06-29 22:51:58 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: move spectrogram to cpu when plotting

Browse Source
This commit is contained in:
Santiago Martinez 2023-04-29 15:39:45 +01:00
parent 616a24347d
commit 715db9a110
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
batdetect2/plot.py
View File

@ -61,7 +61,7 @@ def spectrogram(
""" """
# Convert to numpy array if needed # Convert to numpy array if needed
if isinstance(spec, torch.Tensor): if isinstance(spec, torch.Tensor):
spec = spec.numpy() spec = spec.detach().cpu().numpy()
# Remove batch and channel dimensions if present # Remove batch and channel dimensions if present
spec = spec.squeeze() spec = spec.squeeze()