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

Device fixing #3

Browse Source
This commit is contained in:
mbsantiago 2025-08-25 23:08:49 +01:00
parent d0bab60bf3
commit 9d4a9fc35c
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/batdetect2/postprocess/extraction.py
View File

@ -47,8 +47,8 @@ def extract_prediction_tensor(
indexing="ij",
)
freqs = freqs.flatten().to(detection_heatmap)
times = times.flatten().to(detection_heatmap)
freqs = freqs.flatten().to(detection_heatmap.device)
times = times.flatten().to(detection_heatmap.device)
output_size_preds = output.size_preds.detach()
output_features = output.features.detach()

4
src/batdetect2/train/labels.py
View File

@ -210,8 +210,8 @@ def generate_heatmaps(
indexing="ij",
)
freqs = freqs.to(spec)
times = times.to(spec)
freqs = freqs.to(spec.device)
times = times.to(spec.device)
for sound_event_annotation in clip_annotation.sound_events:
geom = sound_event_annotation.sound_event.geometry