Commit Graph

23 Commits

Author SHA1 Message Date
2720887f88 Added cli for configuring/launching experiments 2024-05-23 11:06:11 +01:00
9536beea04 Refactored to lay groundwork for TUI 2024-05-23 09:50:01 +01:00
02b9e43f4d Made loggers togglable from top level 2024-05-23 09:21:48 +01:00
1be0fa1020 Added overview and plan for expt 2 2024-05-22 15:40:24 +01:00
c3dba79a0b Refactored slightly to allow other experiments 2024-05-22 10:35:14 +01:00
ae14a1d7c0 Added a garbage semantic function
Added a semantic function where the injected "knowledge" is just random
garbage. This function was written to isolate the "knowledge" component
of the other semantic functions, basically to ensure it's the matrices
and not the rest of the process that is making the difference in
training.
2024-05-15 19:01:35 +01:00
e3690b0425 Added wandb.finish to properly start new runs 2024-05-15 18:40:20 +01:00
8a87e864fb Fixed wandb logging mistake 2024-05-15 18:39:42 +01:00
4212c543f8 Update before first push to remote 2024-05-15 16:38:40 +01:00
15a57cc229 Added wandb logging 2024-05-15 16:37:53 +01:00
6600a79f71 Added more varied semantic loss functions 2024-05-15 16:37:53 +01:00
01127de4b3 Added testing step at end of training 2024-05-15 16:37:53 +01:00
62649a39da Modified model to use SGD for less optimal training
There's no point having a loss function testing sandbox where the baseling trains perfectly in 2 epochs. I've DEoptimised the training in this commit to ensure the testing sandbox is actually useful.
2024-05-15 16:37:53 +01:00
389c47ef28 Fixed similarity matrix normalization 2024-05-15 16:37:53 +01:00
66b23a8e76 Minimized model and prepared for testing new loss functions 2024-05-15 16:37:53 +01:00
708f9952b3 First functional training loop 2024-05-15 16:37:53 +01:00
674175b260 Added trainer and made dataset hot-swappable 2024-05-15 16:37:53 +01:00
63bea4d355 Updated dataloader to default to tensor transform 2024-05-15 16:37:53 +01:00
02d70964e1 Added image transformations to dataset fetching 2024-05-15 16:37:53 +01:00
223b064564 Tweaked dataloader for more reliable loading 2024-05-13 15:38:50 +01:00
8204109561 Created consistent dataloader 2024-05-13 15:29:55 +01:00
6df0fea652 Modifications to env 2024-05-13 14:59:27 +01:00
0782af696b Initial env configuration 2024-05-13 11:41:35 +01:00