Commit Graph

102 Commits

Author SHA1 Message Date
henryruhs 857365770f Make validation step more solid, failed on empty checksums 2025-03-11 14:43:09 +01:00
henryruhs d25f2865a9 Normalize validation output 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 c161da2f25 Adjust naming and typing 2025-03-11 14:43:09 +01:00
henryruhs 39818a16df Fix CI 2025-03-11 14:43:09 +01:00
henryruhs e1ba81f220 Improve generator namings, Flip args to source then target 2025-03-11 14:43:09 +01:00
henryruhs bf696be097 Improve generator namings, Flip args to source then target 2025-03-11 14:43:09 +01:00
henryruhs f63bc788ac Fix typo 2025-03-11 14:43:09 +01:00
henryruhs 11bb9065ba Uniform resume checkpoint approach 2025-03-11 14:43:09 +01:00
harisreedhar 7b2b8f0f85 fix discriminator training 2025-03-11 14:43:09 +01:00
henryruhs 999f2c9cbe Rename variable of EmbeddingConverterTrainer 2025-03-11 14:43:09 +01:00
henryruhs 254f3efe68 Remove live portrait submodule 2025-03-11 14:43:09 +01:00
henryruhs 5bb41ecbb2 Adjust README 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 cd4b10c832 Fix imports and update dependencies 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 aa5094e576 We need MS license 2025-03-11 14:43:09 +01:00
harisreedhar 8f8dfecdbc some fixes 2025-03-11 14:43:09 +01:00
harisreedhar 030d912c1b some fixes 2025-03-11 14:43:09 +01:00
henryruhs 0e148845af Extend gitignore 2025-03-11 14:43:09 +01:00
henryruhs e1e0c11bb5 Fix CI 2025-03-11 14:43:09 +01:00
henryruhs 88c4e53192 Use skip_tensor variable 2025-03-11 14:43:09 +01:00
henryruhs 650551c19b Simplify Batch type 2025-03-11 14:43:09 +01:00
henryruhs a971506271 Revert changes in generator 2025-03-11 14:43:09 +01:00
henryruhs b69f69d015 Fix UNet 2025-03-11 14:43:09 +01:00
henryruhs 953525e6b0 Fix typo 2025-03-11 14:43:09 +01:00
henryruhs 58ad6af619 Add Sample suffix again 2025-03-11 14:43:09 +01:00
henryruhs 7264884ff9 Fix CI 2025-03-11 14:43:09 +01:00
henryruhs 1872f99584 Refacto UNet 2025-03-11 14:43:09 +01:00
henryruhs 29e82f909a Move nld to networks 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 3fe32d7832 Fix example config 2025-03-11 14:43:09 +01:00
henryruhs 3be8368eaa Improve lot of types, imports and names 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
henryruhs c6d16c0cf6 Modernize to use ModuleList, Fix some types 2025-03-11 14:43:09 +01:00
henryruhs dfe7ab3b6f Modernize to use ModuleList, Fix some types 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 34d0bc10ed Fix append 2025-03-11 14:43:09 +01:00
henryruhs b785525f3b Fix append 2025-03-11 14:43:09 +01:00
henryruhs dd320ea5be Fix self 2025-03-11 14:43:09 +01:00
henryruhs 71e0ae34c0 Fix self 2025-03-11 14:43:09 +01:00