Commit Graph

99 Commits

Author SHA1 Message Date
henryruhs 094d5cea9e Fix CI 2025-03-11 14:43:09 +01:00
henryruhs f6c59257d9 Partial use Resnet34 as a DownSample replacement 2025-03-11 14:43:09 +01:00
henryruhs 83ef075b1d Remove map_location 2025-03-11 14:43:09 +01:00
henryruhs d153c68813 Better split_ratio 2025-03-11 14:43:09 +01:00
henryruhs 00dccf07b9 Fix tensor foobar 2025-03-11 14:43:09 +01:00
henryruhs 5a6e3393e2 Revert loss for the moment 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 84503761b9 follow the not invented here syndrome 2025-03-11 14:43:09 +01:00
henryruhs 723e9fde78 follow the not invented here syndrome 2025-03-11 14:43:09 +01:00
henryruhs 09e913233b Follow convention of the other project 2025-03-11 14:43:09 +01:00
henryruhs b4bbd862e2 Follow convention of the other project 2025-03-11 14:43:09 +01:00
harisreedhar 4d8433f54a changes 2025-03-11 14:43:09 +01:00
harisreedhar db44c91dd8 changes 2025-03-11 14:43:09 +01:00
henryruhs b33281425a This is not a TensorDataset 2025-03-11 14:43:09 +01:00
henryruhs 04eaa831ea Rename to resume_path only 2025-03-11 14:43:09 +01:00
henryruhs da51c5336d This can be marked static 2025-03-11 14:43:09 +01:00
henryruhs 9bd68c3d14 Similarity validation for embedding converter 2025-03-11 14:43:09 +01:00
henryruhs 0a50e2d706 Rename data loader 2025-03-11 14:43:09 +01:00
henryruhs 40dcef7fc7 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 39c0313202 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
henryruhs dcf19634d1 Brain fart 2025-03-11 14:43:09 +01:00
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 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 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 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 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