Commit Graph

30 Commits

Author SHA1 Message Date
henryruhs 5b3b2abdd7 Remove annoying Tuner, Match both trainer configs a bit more 2025-03-11 14:43:10 +01:00
henryruhs ee3fc40e83 Move output channels to config 2025-03-11 14:43:10 +01:00
henryruhs 14bbece850 Rename id_ to identity_ 2025-03-11 14:43:10 +01:00
henryruhs 6eabcad1d0 Introduce new AttributeLoss class 2025-03-11 14:43:10 +01:00
henryruhs 086d9eed87 Introduce new ReconstructionLoss class 2025-03-11 14:43:09 +01:00
henryruhs 085c493e18 Rename id_embedder to embedder, Tons of naming in training step, Introduce new IdentityLoss class 2025-03-11 14:43:09 +01:00
henryruhs dc0abff0ce Minor typo 2025-03-11 14:43:09 +01:00
henryruhs 10ce04ed58 Introduce feature flag for Unet 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 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
harisreedhar db44c91dd8 changes 2025-03-11 14:43:09 +01:00
henryruhs 04eaa831ea Rename to resume_path only 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
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 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
harisreedhar 8f8dfecdbc some fixes 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 860771e482 Fix weight identity 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
harisreedhar 66af7d7957 add face swapper preview 2025-03-11 14:43:03 +01:00
harisreedhar dc0ef53668 add face swapper preview 2025-03-11 14:43:03 +01:00
henryruhs a1cd025f81 Mixed bag of cleanups 2025-03-11 14:43:03 +01:00
henryruhs 9c15f584aa Mixed bag of cleanups 2025-03-11 14:43:03 +01:00
harisreedhar fcb3390796 again again cleaning 2025-03-11 14:43:03 +01:00