Commit Graph

420 Commits

Author SHA1 Message Date
mbsantiago
c73984b213 Small fixes 2025-09-08 18:35:02 +01:00
mbsantiago
d8d2e5a2c2 Remove preprocessing modules 2025-09-08 18:11:58 +01:00
mbsantiago
b056d7d28d Make sure training is still working 2025-09-08 18:03:56 +01:00
mbsantiago
95a884ea16 Update tests 2025-09-08 18:00:17 +01:00
mbsantiago
b7ae526071 Big changes in data module 2025-09-08 17:50:25 +01:00
mbsantiago
cf6d0d1ccc Remove stale tests 2025-09-07 11:03:46 +01:00
mbsantiago
709b6355c2 torch.multiprocessing didn't work, returning to serial processing 2025-09-01 11:27:23 +01:00
mbsantiago
db2ad11743 Make matching in parallel for speedup 2025-09-01 11:19:02 +01:00
mbsantiago
e0ecc3c3d1 Clear evaluation callback after epoch ends 2025-09-01 08:56:38 +01:00
mbsantiago
71c2301c21 Independent preprocessor for generating validation plots 2025-08-31 23:04:16 +01:00
mbsantiago
d3d2a28130 Move detections array to cpu 2025-08-31 22:59:06 +01:00
mbsantiago
5b9a5a968f Refactor eval code 2025-08-31 22:57:02 +01:00
mbsantiago
356be57f62 Update to latests soundevent 2025-08-31 20:17:48 +01:00
mbsantiago
8d093c3ca2 Use custom AugmentationSequence instead of nn.Sequential 2025-08-31 19:27:15 +01:00
mbsantiago
2f4edeffff Create separate preprocessor for the train/val datasets 2025-08-31 19:22:45 +01:00
mbsantiago
cca1d82d63 Remove blocking .to(device) 2025-08-31 19:10:06 +01:00
mbsantiago
55f473c9ca Fix logging issue 2025-08-31 19:06:09 +01:00
mbsantiago
40f6b64611 Remove train preprocessing 2025-08-31 18:28:52 +01:00
mbsantiago
1cec332dd5 Change default train duration to 0.256 instead of 0.512 2025-08-30 14:08:00 +01:00
mbsantiago
93e89ecc46 LR Scheduler takes num of total batches 2025-08-28 08:52:11 +01:00
mbsantiago
34ef9e92a1 Make sure preprocessing is batchable 2025-08-27 23:58:38 +01:00
mbsantiago
0b5ac96fe8 Update model config 2025-08-27 23:58:07 +01:00
mbsantiago
dba6d2d918 Updating configs 2025-08-27 23:44:49 +01:00
mbsantiago
ff754a1269 Tweaks of augmentation config 2025-08-27 18:23:38 +01:00
mbsantiago
ed76ec24b6 Plot anchor points 2025-08-27 18:13:40 +01:00
mbsantiago
d25efdad10 Fix plotting after update 2025-08-26 11:48:06 +01:00
mbsantiago
3043230d4f Device fixing #5 2025-08-25 23:20:30 +01:00
mbsantiago
67e37227f5 Device fixing #4 2025-08-25 23:11:33 +01:00
mbsantiago
9d4a9fc35c Device fixing #3 2025-08-25 23:08:49 +01:00
mbsantiago
d0bab60bf3 Device fixing #2 2025-08-25 23:07:09 +01:00
mbsantiago
a267db290c Device fixing 2025-08-25 23:04:13 +01:00
mbsantiago
441ccb3382 Remove user warning from plotting function 2025-08-25 22:49:48 +01:00
mbsantiago
281c4dcb8a Remove xr from postprocess 2025-08-25 22:46:21 +01:00
mbsantiago
cc9e47b022 Fix plotting after changes 2025-08-25 19:07:12 +01:00
mbsantiago
1f26103f42 Cleanup train preprocessing 2025-08-25 18:37:46 +01:00
mbsantiago
c80078feee Removing stale tests 2025-08-25 17:23:27 +01:00
mbsantiago
0bb0caddea Update augmentations 2025-08-25 17:06:17 +01:00
mbsantiago
76dda0a0e9 Fix train preprocessing 2025-08-25 14:01:31 +01:00
mbsantiago
c36ef3ecb5 Labels to torch 2025-08-25 12:43:34 +01:00
mbsantiago
667b18a54d Preprocessing in pytorch 2025-08-25 11:41:55 +01:00
mbsantiago
61115d562c Moved types to dedicated module 2025-08-24 10:55:48 +01:00
mbsantiago
02adc19070 Better structure for training module 2025-08-23 18:23:45 +01:00
mbsantiago
d4f249366e Clean up the detector head 2025-08-18 16:07:51 +01:00
mbsantiago
98f83e8b34 Small fix 2025-08-18 10:25:14 +01:00
mbsantiago
9d1497b3f4 Handle case of empty preds or gt 2025-08-17 23:00:40 +01:00
mbsantiago
7af72912da Implement previous matching strategy (greedy) 2025-08-17 22:52:00 +01:00
mbsantiago
4aea3fb2b0 Copy librosa PCEN implementation 2025-08-17 18:02:40 +01:00
mbsantiago
c1945ebdb7 Minor difference in loss fixed 2025-08-16 18:49:30 +01:00
mbsantiago
13da5d2814 Remove unnecessary extra conv layer 2025-08-16 18:32:35 +01:00
mbsantiago
78a0975864 Same issue 2025-08-14 11:07:27 +01:00