Commit Graph

55 Commits

Author SHA1 Message Date
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
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
henryruhs da51c5336d This can be marked static 2025-03-11 14:43:09 +01:00
henryruhs c041073953 Modernize data loader, remove read image helper 2025-03-11 14:43:09 +01:00
henryruhs 354315502b Modernize data loader, remove read image helper 2025-03-11 14:43:09 +01:00
henryruhs 254bc17c98 Modernize data loader, remove read image helper 2025-03-11 14:43:09 +01:00
henryruhs 3cf9711df0 Modernize data loader, remove read image helper 2025-03-11 14:43:09 +01:00
harisreedhar d1bf54276d changes 2025-03-11 14:43:09 +01:00
harisreedhar 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 777a8384c2 Extend README with TensorBoard guide 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 0b7d25a36e Adjust licenses 2025-03-11 14:43:09 +01:00
henryruhs 026bcf0c97 Adjust licenses 2025-03-11 14:43:09 +01:00
henryruhs e1e0c11bb5 Fix CI 2025-03-11 14:43:09 +01:00
henryruhs 650551c19b Simplify Batch type 2025-03-11 14:43:09 +01:00
henryruhs 257ab668ee Name what it is 2025-03-11 14:43:09 +01:00
henryruhs 62d897f9d8 Direct return layers 2025-03-11 14:43:09 +01:00
henryruhs f05ff6cdb1 Enforce IR version for older onnxruntime 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 e33bc0d52a Modernize to use ModuleList, Fix some types 2025-03-11 14:43:09 +01:00
henryruhs f3409c5ade Modernize to use ModuleList, Fix some types 2025-03-11 14:43:09 +01:00