Commit Graph

431 Commits

Author SHA1 Message Date
mbsantiago
6d70140bc9 Default to normal anchor 2025-09-13 13:56:47 +01:00
mbsantiago
4fd2e84773 Fix clip missalignment in validation dataset 2025-09-11 11:09:59 +01:00
mbsantiago
74c419f674 Update default config 2025-09-10 21:49:57 +01:00
mbsantiago
e65d5a6846 Added more clipping options for validation 2025-09-10 21:09:51 +01:00
mbsantiago
615c811bb4 Add detection_class_name to targets protocol 2025-09-09 20:30:20 +01:00
mbsantiago
41b18c3f0a Better order for checkpoints 2025-09-09 15:56:46 +01:00
mbsantiago
16a0fa7b75 Add targets to train cli 2025-09-09 15:45:00 +01:00
mbsantiago
115084fd2b Updat lightning version 2025-09-09 15:31:40 +01:00
mbsantiago
951dc59718 Add seed option to train 2025-09-09 13:23:56 +01:00
mbsantiago
3376be06a4 Add experiment name 2025-09-09 09:02:25 +01:00
mbsantiago
cd4955d4f3 Eval 2025-09-08 22:04:30 +01:00
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