Commit Graph

230 Commits

Author SHA1 Message Date
harisreedhar f2d3f8a19f changes 2025-03-11 14:43:10 +01:00
henryruhs ceb3c0cfdf Rename to AugmentTransform 2025-03-11 14:43:10 +01:00
henryruhs dd0a2fe649 Add albumentations 2025-03-11 14:43:10 +01:00
henryruhs e5a4a54e61 Introduce Transform classes, Add albumentations 2025-03-11 14:43:10 +01:00
henryruhs 9dc1031fa5 Fix same and equal 2025-03-11 14:43:10 +01:00
henryruhs d93daf9e5f Introduce batch mode via config for equal and same batches 2025-03-11 14:43:10 +01:00
henryruhs 3a61da8bab Introduce batch mode via config for equal and same batches 2025-03-11 14:43:10 +01:00
harisreedhar 83c20f8331 changes 2025-03-11 14:43:10 +01:00
harisreedhar dfd9e99aed changes 2025-03-11 14:43:10 +01:00
harisreedhar 2fb0b4289d changes 2025-03-11 14:43:10 +01:00
harisreedhar e6ea454360 supress warning 2025-03-11 14:43:10 +01:00
henryruhs 80d1694e23 Replace naming warp_matrix with warp_template 2025-03-11 14:43:10 +01:00
henryruhs 5e68de9170 Add config to dataloader alignment 2025-03-11 14:43:10 +01:00
henryruhs 6ca68f1408 Add config to dataloader alignment 2025-03-11 14:43:10 +01:00
henryruhs 43f99db5d7 Add config to dataloader alignment 2025-03-11 14:43:10 +01:00
henryruhs c9e70ebc18 Add config to dataloader alignment 2025-03-11 14:43:10 +01:00
henryruhs 1ab57b1d3f Add config to dataloader alignment 2025-03-11 14:43:10 +01:00
henryruhs cc9a0ba83e Add config to dataloader alignment 2025-03-11 14:43:10 +01:00
henryruhs 589568bfb5 Add config to dataloader alignment 2025-03-11 14:43:10 +01:00
harisreedhar 34a7f3ef55 changes 2025-03-11 14:43:10 +01:00
harisreedhar c35b0a6f4c changes 2025-03-11 14:43:10 +01:00
harisreedhar 23ab7dc89d fix 2025-03-11 14:43:10 +01:00
henryruhs 8443f3512a torch.float32 is the default 2025-03-11 14:43:10 +01:00
henryruhs d3a2035d7a Fix order 2025-03-11 14:43:10 +01:00
henryruhs 16c8b32269 Fix some naming and types 2025-03-11 14:43:10 +01:00
harisreedhar 65ab796835 changes 2025-03-11 14:43:10 +01:00
henryruhs 56be3f0b9b Rename to validation_score 2025-03-11 14:43:10 +01:00
henryruhs a22adaf51f Adjust some namings 2025-03-11 14:43:10 +01:00
henryruhs 0055c0c97f Adjust some namings 2025-03-11 14:43:10 +01:00
harisreedhar 35b779b1ed change pretrained models mode to eval 2025-03-11 14:43:10 +01:00
harisreedhar ea1b0205f0 constructor injection 2025-03-11 14:43:10 +01:00
harisreedhar a5eb7d6aa1 change Adam to AdamW 2025-03-11 14:43:10 +01:00
harisreedhar b27b8663e5 changes 2025-03-11 14:43:10 +01:00
harisreedhar d87f6c0b15 changes 2025-03-11 14:43:10 +01:00
harisreedhar 5d1b90ff19 changes 2025-03-11 14:43:10 +01:00
harisreedhar 2ddcf52b66 changes 2025-03-11 14:43:10 +01:00
henryruhs c8801ececd Fix export using Trainer 2025-03-11 14:43:10 +01:00
henryruhs 84b4451366 Fix export using Trainer 2025-03-11 14:43:10 +01:00
henryruhs cadbe9cf76 Follow the concept of layers and sequences 2025-03-11 14:43:10 +01:00
henryruhs 58a85a80bb Use high float32 matmul precision 2025-03-11 14:43:10 +01:00
harisreedhar ab0a59fb74 changes 2025-03-11 14:43:10 +01:00
henryruhs 578b07a7f4 Add StatefulDataloader, Manual trigger scheduler 2025-03-11 14:43:10 +01:00
henryruhs 7ce9d27097 Different naming 2025-03-11 14:43:10 +01:00
henryruhs 8c24c9ec27 Fix CI 2025-03-11 14:43:10 +01:00
henryruhs 0ad2556c4c Improve optimizer configs 2025-03-11 14:43:10 +01:00
henryruhs 484a49c27d Restore dataset behaviour for same person 2025-03-11 14:43:10 +01:00
harisreedhar e8cc2bfff1 add gaze loss 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