Commit Graph

325 Commits

Author SHA1 Message Date
mbsantiago
e8db1d4050 Fix hyperparameter saving 2025-06-26 19:53:19 -06:00
mbsantiago
b396d4908a Fix trainer init 2025-06-26 19:40:27 -06:00
mbsantiago
ce15a0f152 Fix trainer init 2025-06-26 19:35:23 -06:00
mbsantiago
16febed792 Add test that ensures example_config is valid 2025-06-26 19:21:46 -06:00
mbsantiago
d67ae9be05 Update config to remove optimiser level 2025-06-26 18:42:53 -06:00
mbsantiago
15de168a20 Update build trainer 2025-06-26 17:43:56 -06:00
mbsantiago
0c8fae4a72 Instantiate lightnign module from config 2025-06-26 17:39:50 -06:00
mbsantiago
6d57f96c07 update gitignore 2025-06-26 16:19:33 -06:00
mbsantiago
a0622aa9a4 Fix train issues 2025-06-26 16:16:29 -06:00
mbsantiago
587742b41e Change train to use full config 2025-06-26 16:02:41 -06:00
mbsantiago
6d91153a56 use just instead of make 2025-06-26 13:29:23 -06:00
mbsantiago
70c96b6844 Move configs to example_data folder 2025-06-26 13:29:13 -06:00
mbsantiago
22f7d46f46 Improve logging of train preprocessing 2025-06-26 13:08:44 -06:00
mbsantiago
1384c549f7 Create TrainPreprocessConfig 2025-06-26 12:30:16 -06:00
mbsantiago
4b6acd5e6e Add manual logging of hyperparams 2025-06-26 11:59:33 -06:00
mbsantiago
2ac968d65b Test with dvc live 2025-06-26 10:05:01 -06:00
mbsantiago
166dad20bd Rename BlockGroupConfig to LayerGroupConfig 2025-06-26 10:04:42 -06:00
mbsantiago
cbb02cf69e Add dvclive as an optional group 2025-06-26 10:04:16 -06:00
mbsantiago
647468123e Update model config in example config 2025-06-26 10:04:00 -06:00
mbsantiago
363eb9fb2f Add example-train to makefile 2025-06-26 10:03:52 -06:00
mbsantiago
b21c224985 Add example preprocess to make file 2025-06-26 07:55:40 -06:00
mbsantiago
152a577511 Add comprehensive conf file 2025-06-26 07:55:33 -06:00
mbsantiago
136949c4e7 Add logging config 2025-06-26 07:55:24 -06:00
mbsantiago
a462beaeb8 Remove rogue print 2025-06-24 06:27:41 -06:00
mbsantiago
daab8ff0d7 Fix validation is_in_subclip after encoder changes 2025-06-24 06:26:08 -06:00
mbsantiago
235f0e27da Add load dataset config function 2025-06-24 12:40:29 +01:00
mbsantiago
b5b4229990 Fix testing issues 2025-06-23 19:08:55 +01:00
mbsantiago
8253b5bdc4 Update makefile 2025-06-23 18:53:15 +01:00
mbsantiago
c7ea361cf4 Implement changes needed to make roi encode/decode class dependent 2025-06-23 18:52:36 +01:00
mbsantiago
3407e1b5f0 Add other roi tests 2025-06-21 23:51:07 +01:00
mbsantiago
0a0d6f7162 Use standard dependency-groups instead of tool.uv section 2025-06-21 23:01:54 +01:00
mbsantiago
ad0f0bcb24 Add tests for peak energy function 2025-06-21 23:01:08 +01:00
mbsantiago
3103630c26 Update pyproject to use src layout 2025-06-21 13:49:06 +01:00
mbsantiago
960558be8b move to src layout 2025-06-21 13:48:40 +01:00
mbsantiago
e352dc40bd Fixed Target object after changes to roi 2025-06-21 13:47:04 +01:00
mbsantiago
c559bcc682 Changed ROIMapper protocol to only have encoder/decoder methods 2025-06-21 11:44:15 +01:00
mbsantiago
ebad489cb1 Merge branch 'train' of github.com:macaodha/batdetect2 into train 2025-06-20 16:03:15 +01:00
mbsantiago
3c9e5aca2f Fix testing issues 2025-06-20 15:57:11 +01:00
Santiago Martinez Balvanera
a62f07ebdd to cpu 2025-06-19 00:46:43 +01:00
Santiago Martinez Balvanera
434fc652a2 Add collate fn 2025-06-19 00:46:34 +01:00
mbsantiago
84a13c65a7 Add iterators module 2025-06-19 00:28:24 +01:00
mbsantiago
ebc89af4c6 Use soundevent.to_db 2025-06-19 00:28:01 +01:00
mbsantiago
835fe1ccdf Remove old requirements files 2025-05-16 08:55:38 +01:00
mbsantiago
b819180656 Add preprocess command to init 2025-05-01 07:15:35 +01:00
mbsantiago
fdab0860fd Add num workers to cli 2025-04-30 23:40:26 +01:00
mbsantiago
3b5623ddca create train cli command 2025-04-30 23:27:38 +01:00
mbsantiago
2913fa59a4 Fix tests 2025-04-30 22:51:49 +01:00
mbsantiago
9c8b8fb200 Create metrics 2025-04-30 22:51:33 +01:00
mbsantiago
bc86c94f8e Adding evaluation callback 2025-04-25 17:12:57 +01:00
mbsantiago
9106b9f408 Removing stale functions from legacy module 2025-04-25 17:12:39 +01:00