Commit Graph

84 Commits

Author SHA1 Message Date
henryruhs 803902c8bb Add sync_dist to validation score logging 2025-03-16 09:01:40 +01:00
henryruhs 20aa5114e1 Extend config 2025-03-12 16:24:03 +01:00
henryruhs 2af4f2c8ab Extend config 2025-03-12 16:21:47 +01:00
henryruhs 8101b15e1c Revert the config dicts 2025-03-11 14:43:10 +01:00
henryruhs 1dfd230fc5 Revert the config dicts 2025-03-11 14:43:10 +01:00
henryruhs a5d99c139e Fix bug 2025-03-11 14:43:10 +01:00
henryruhs a0c42bedbe Improve imports 2025-03-11 14:43:10 +01:00
henryruhs d215b6f98b Fix CI 2025-03-11 14:43:10 +01:00
henryruhs 01278d679f Forward config parser 2025-03-11 14:43:10 +01:00
henryruhs cf26f66e36 More adjustments 2025-03-11 14:43:10 +01:00
henryruhs f4a1e18ca9 More adjustments 2025-03-11 14:43:10 +01:00
henryruhs 7ab7efbbf4 More adjustments 2025-03-11 14:43:10 +01:00
henryruhs 6a5f81e5fe More adjustments 2025-03-11 14:43:10 +01:00
henryruhs 6a11603e7e More adjustments 2025-03-11 14:43:10 +01:00
henryruhs ff9b777b28 More adjustments 2025-03-11 14:43:10 +01:00
henryruhs 5bacb048dd More adjustments 2025-03-11 14:43:10 +01:00
henryruhs 847579f925 More adjustments 2025-03-11 14:43:10 +01:00
henryruhs 57aad5204e More adjustments 2025-03-11 14:43:10 +01:00
henryruhs d3b0051912 More adjustments 2025-03-11 14:43:10 +01:00
henryruhs d944d95bfd Fix CI 2025-03-11 14:43:10 +01:00
henryruhs b59e172fa3 Always config injection 2025-03-11 14:43:10 +01:00
henryruhs 368da824aa Fix CI 2025-03-11 14:43:10 +01:00
henryruhs e61e470432 Fix CI 2025-03-11 14:43:10 +01:00
henryruhs c8953ce8a1 Apply new config approach for embedding converter 2025-03-11 14:43:10 +01:00
henryruhs 56be3f0b9b Rename to validation_score 2025-03-11 14:43:10 +01:00
henryruhs 84b4451366 Fix export using Trainer 2025-03-11 14:43:10 +01:00
henryruhs 58a85a80bb Use high float32 matmul precision 2025-03-11 14:43:10 +01:00
henryruhs 578b07a7f4 Add StatefulDataloader, Manual trigger scheduler 2025-03-11 14:43:10 +01:00
henryruhs 0ad2556c4c Improve optimizer configs 2025-03-11 14:43:10 +01:00
henryruhs a951d700fc Restore RGB normalization 2025-03-11 14:43:10 +01:00
henryruhs 6eff69a41a There is no need to make directories 2025-03-11 14:43:10 +01:00
henryruhs 5b3b2abdd7 Remove annoying Tuner, Match both trainer configs a bit more 2025-03-11 14:43:10 +01:00
henryruhs bfcbd6bf95 Fix CI 2025-03-11 14:43:10 +01:00
henryruhs 607c55ff1f Remove Numpy and CV2 to fully use Tensors 2025-03-11 14:43:10 +01:00
henryruhs 93cbbf52d0 Fix CI 2025-03-11 14:43:10 +01:00
henryruhs f5cd6b6336 This should be a StaticDataset, Fix learning rate finder 2025-03-11 14:43:10 +01:00
henryruhs 257e5e56a4 Rename split ratio 2025-03-11 14:43:10 +01:00
henryruhs f19908ccd6 Fix spacing for lambda 2025-03-11 14:43:10 +01:00
henryruhs 18a2531b54 Fix naming in dataset.py 2025-03-11 14:43:10 +01:00
henryruhs 14b9bccafe Introduce new DiscriminatorLoss class, Remove useless super call params 2025-03-11 14:43:10 +01:00
henryruhs 579d3ef51c Introduce new GazeLoss class (switched to mean) 2025-03-11 14:43:10 +01:00
henryruhs 6eabcad1d0 Introduce new AttributeLoss class 2025-03-11 14:43:10 +01:00
henryruhs a6e1405c70 Restore map_location = 'cpu' 2025-03-11 14:43:09 +01:00
henryruhs ac41bab3a2 Restore map_location = 'cpu' 2025-03-11 14:43:09 +01:00
henryruhs 83ef075b1d Remove map_location 2025-03-11 14:43:09 +01:00
henryruhs 575f215408 lr only in optimizer 2025-03-11 14:43:09 +01:00
henryruhs c17378f3c7 Let's call it batch ratio 2025-03-11 14:43:09 +01:00
henryruhs 3d9ff4add0 Adjust validation to one time per epoch 2025-03-11 14:43:09 +01:00
henryruhs 5934b47961 Add cosine_similarity 2025-03-11 14:43:09 +01:00
henryruhs 04eaa831ea Rename to resume_path only 2025-03-11 14:43:09 +01:00