Commit Graph
32 Commits
Author SHA1 Message Date
henryruhs 56be3f0b9b Rename to validation_score 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 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 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 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 575f215408 lr only in optimizer 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
henryruhs c1bed34c27 More tweaks 2025-03-11 14:43:09 +01:00
henryruhs 4078681031 Fix typing 2025-03-11 14:43:09 +01:00
henryruhs 15ee6fa763 Simplify sizes 2025-03-11 14:43:09 +01:00
henryruhs 251e610f0e Follow the lightning naming and call this dataset, Improve config and types 2025-03-11 14:43:09 +01:00
harisreedharandhenryruhs d1bf54276d changes 2025-03-11 14:43:09 +01:00
harisreedharandhenryruhs b47c6b72ee changes 2025-03-11 14:43:09 +01:00
henryruhs 0d45568bd1 Uniform validation, Add cosine_similarity validation to face swapper 2025-03-11 14:43:09 +01:00
henryruhs ccf6fa7f43 Uniform validation, Add cosine_similarity validation to face swapper 2025-03-11 14:43:09 +01:00
henryruhs 28977d37d6 Uniform validation, Add cosine_similarity validation to face swapper 2025-03-11 14:43:09 +01:00
henryruhs 11bb9065ba Uniform resume checkpoint approach 2025-03-11 14:43:09 +01:00
henryruhs 999f2c9cbe Rename variable of EmbeddingConverterTrainer 2025-03-11 14:43:09 +01:00
henryruhs 0949c1358b Fix CI 2025-03-11 14:43:09 +01:00
henryruhs d9e10a9f7c Use lightning over pytorch_lightning import, Configure tensorboard 2025-03-11 14:43:09 +01:00
henryruhs b6b4f9f65b Improve lot of types, imports and names 2025-03-11 14:43:09 +01:00
henryruhs 3c6dfa4efe Modernize to use ModuleList, Fix some types 2025-03-11 14:43:09 +01:00
henryruhs 746cc86d52 Fix monitor 2025-03-11 14:43:09 +01:00
henryruhs 1b6e7a6ca5 Rename ArcFace Converter to Embedding Converter, Add EmbeddingDataset, Add learning rate to config 2025-03-11 14:43:06 +01:00