Commit Graph

209 Commits

Author SHA1 Message Date
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
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 84be7d1ffb Remove deprecated argument 2025-03-11 14:43:10 +01:00
henryruhs 8b53c76a0a Remove deprecated argument 2025-03-11 14:43:10 +01:00
henryruhs 7d8cb146a4 Use latest numpy 2025-03-11 14:43:10 +01:00
henryruhs 303cbfa024 Final refactoring for AAD done 2025-03-11 14:43:10 +01:00
henryruhs bc174186eb Final refactoring for AAD done 2025-03-11 14:43:10 +01:00
henryruhs d7158749c2 Improve naming 2025-03-11 14:43:10 +01:00
henryruhs 8b2b6892aa Improve naming 2025-03-11 14:43:10 +01:00
henryruhs 5bba2a1c69 Remove the condition from reconstruction loss 2025-03-11 14:43:10 +01:00
henryruhs 94480e16eb Rename to AAD as this is not a full AIENet 2025-03-11 14:43:10 +01:00
henryruhs bbcb1c35f0 Modernize AIENet 2025-03-11 14:43:10 +01:00
henryruhs ee3fc40e83 Move output channels to config 2025-03-11 14:43:10 +01:00
henryruhs 335d597e53 Rename to aienet.py 2025-03-11 14:43:10 +01:00
henryruhs 14bbece850 Rename id_ to identity_ 2025-03-11 14:43:10 +01:00
henryruhs fad38da864 Rename temp1 with positive and temp2 with negative 2025-03-11 14:43:10 +01:00