Commit Graph

54 Commits

Author SHA1 Message Date
mbsantiago
5cc5767eff fix: rename detector heads and refresh bundled checkpoint 2026-05-06 12:50:53 +01:00
mbsantiago
a27d1bbfd3 refactor: derive training config from the model 2026-05-06 12:48:40 +01:00
mbsantiago
7c05fb8577 feat: default to bundled checkpoint
Fall back to the bundled uk_same model when no checkpoint is provided in the shared loader and fine-tune CLI. Keep tests aligned with the new default resolution behavior.
2026-05-06 10:33:04 +01:00
mbsantiago
84918086c8 feat: streamline bundled checkpoint handling
Support packaged model aliases and save weights-only checkpoints by default so distributed models stay small while remaining easy to load.
2026-05-05 21:34:54 +01:00
mbsantiago
5526ac99fc Remove stale dependencies 2026-05-05 16:20:37 +01:00
mbsantiago
f5afa9881c feat: load checkpoints from Hugging Face 2026-05-05 15:46:39 +01:00
mbsantiago
5a974711b0 feat: log training provenance artifacts 2026-05-05 14:09:53 +01:00
mbsantiago
7b2699786f feat: add checkpoint finetuning workflow 2026-05-05 12:16:37 +01:00
mbsantiago
7a10b7ffff refactor: remove aggregate app config 2026-05-04 23:10:31 +01:00
mbsantiago
aa36df668f feat: persist target configs in checkpoints 2026-05-04 22:54:32 +01:00
mbsantiago
57236fc82a refactor: decouple model metadata from target configs 2026-05-04 21:18:17 +01:00
mbsantiago
e80fe8675d Mark some tests as slow for quicker feedback 2026-03-29 15:55:24 +01:00
mbsantiago
2b235e28bb Separate targets and ROI 2026-03-29 13:11:42 +01:00
mbsantiago
a1fad6d7d7 Add test for checkpointing 2026-03-19 01:26:37 +00:00
mbsantiago
99b9e55c0e Add extra and strict arguments to load config functions, migrate example config 2026-03-18 22:06:45 +00:00
mbsantiago
bf5b88016a Polish evaluate and train CLI 2026-03-18 19:15:57 +00:00
mbsantiago
9fa703b34b Allow training on existing model 2026-03-18 13:58:52 +00:00
mbsantiago
751be53edf Moving types around to each submodule 2026-03-18 00:01:35 +00:00
mbsantiago
feee2bdfa3 Add scheduler and optimizer module 2026-03-17 21:16:41 +00:00
mbsantiago
615c7d78fb Added a full test of training and saving 2026-03-17 15:38:07 +00:00
mbsantiago
56f6affc72 use train config in training module 2026-03-17 14:16:40 +00:00
mbsantiago
65bd0dc6ae Restructure model config 2026-03-17 13:33:13 +00:00
mbsantiago
202c6cbab0 Fix tests 2025-11-28 18:10:37 +00:00
mbsantiago
8366410332 Update test 2025-10-27 19:39:38 +00: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
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
34ef9e92a1 Make sure preprocessing is batchable 2025-08-27 23:58:38 +01:00
mbsantiago
281c4dcb8a Remove xr from postprocess 2025-08-25 22:46:21 +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
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
59aaf07af5 Update tests after incorporating term registry from soundevent 2025-08-12 18:44:18 +01:00
mbsantiago
e8db1d4050 Fix hyperparameter saving 2025-06-26 19:53:19 -06:00
mbsantiago
16febed792 Add test that ensures example_config is valid 2025-06-26 19:21:46 -06:00
mbsantiago
0c8fae4a72 Instantiate lightnign module from config 2025-06-26 17:39:50 -06:00
mbsantiago
c7ea361cf4 Implement changes needed to make roi encode/decode class dependent 2025-06-23 18:52:36 +01:00
mbsantiago
c559bcc682 Changed ROIMapper protocol to only have encoder/decoder methods 2025-06-21 11:44:15 +01:00
mbsantiago
3c9e5aca2f Fix testing issues 2025-06-20 15:57:11 +01:00
mbsantiago
2913fa59a4 Fix tests 2025-04-30 22:51:49 +01:00
mbsantiago
2a45859393 Run lint fix 2025-04-23 23:19:08 +01:00
mbsantiago
59bd14bc79 Added clips for random cliping and augmentations 2025-04-23 23:15:08 +01:00
mbsantiago
541be15c9e Update augmentation tests to new structure 2025-04-22 09:00:57 +01:00
mbsantiago
257e1e01bf Fixing small errors in tests 2025-04-22 08:51:21 +01:00
mbsantiago
7c89e82579 Fixing imports after restructuring 2025-04-22 00:36:34 +01:00
mbsantiago
f9e005ec8b Add tests for aoef loading 2025-04-18 13:32:50 +01:00