423 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
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
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