Commit Graph

48 Commits

Author SHA1 Message Date
harisreedhar 4f5ac00a7b changes 2025-04-08 12:29:07 +05:30
harisreedhar 4b851a173d changes 2025-03-28 16:32:44 +05:30
harisreedhar c1f39a73dd change face-parser to face-masker 2025-03-23 18:18:10 +05:30
henryruhs ad675ae633 Join MaskNet to guide generator 2025-03-16 12:23:08 +01:00
henryruhs 94571c5676 Rename attribute to feature 2025-03-16 08:39:35 +01:00
henryruhs 5550c89f43 Fix AAD naming, Attribute and Embedding 2025-03-12 16:30:59 +01:00
henryruhs 2af4f2c8ab Extend config 2025-03-12 16:21:47 +01:00
henryruhs afab997ffc Rename parser to face parser 2025-03-11 14:43:10 +01:00
harisreedhar f90fd73b54 add strategy to config 2025-03-11 14:43:10 +01:00
harisreedhar d809f66216 changes 2025-03-11 14:43:10 +01:00
harisreedhar 27440ef023 changes 2025-03-11 14:43:10 +01:00
harisreedhar 606bf42089 fix 2025-03-11 14:43:10 +01:00
harisreedhar a3ac4d5ddd changes 2025-03-11 14:43:10 +01:00
harisreedhar 1659805b08 changes 2025-03-11 14:43:10 +01:00
harisreedhar 8f1f002c64 add masknet 2025-03-11 14:43:10 +01:00
harisreedhar 7cf5609c1f changes 2025-03-11 14:43:10 +01:00
henryruhs f9de4ce78a Remove UnetPro, make float values visible in README 2025-03-11 14:43:10 +01:00
henryruhs 5056b8df75 Variable AAD layer according to output size 2025-03-11 14:43:10 +01:00
henryruhs 430c71d031 Adjust config and namings 2025-03-11 14:43:10 +01:00
harisreedhar 176dced1f6 changes 2025-03-11 14:43:10 +01:00
henryruhs 18a605e1a3 Change the weight config order 2025-03-11 14:43:10 +01:00
harisreedhar f2d3f8a19f changes 2025-03-11 14:43:10 +01:00
henryruhs 3a61da8bab Introduce batch mode via config for equal and same batches 2025-03-11 14:43:10 +01:00
harisreedhar dfd9e99aed changes 2025-03-11 14:43:10 +01:00
henryruhs 80d1694e23 Replace naming warp_matrix with warp_template 2025-03-11 14:43:10 +01:00
henryruhs 589568bfb5 Add config to dataloader alignment 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 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 14b9bccafe Introduce new DiscriminatorLoss class, Remove useless super call params 2025-03-11 14:43:10 +01:00
henryruhs 2220f5ef08 Add missing encoder_type to config 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 723e9fde78 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 11bb9065ba Uniform resume checkpoint approach 2025-03-11 14:43:09 +01:00
harisreedhar 030d912c1b 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 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
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
harisreedhar 3b8b6442fc add infer and some cleaning 2025-03-11 14:43:03 +01:00
harisreedhar cfcf0ee2bd add glob image pattern to config 2025-03-11 14:43:03 +01:00
harisreedhar d018dd0633 some cleaning and add exporting 2025-03-11 14:43:03 +01:00
harisreedhar 008a221f55 cleaning 2025-03-11 14:43:03 +01:00
harisreedhar e6c2a64256 ugly training code 2025-03-11 14:43:02 +01:00
harisreedhar 8e53c6bc9f new swapper 2025-03-11 14:43:02 +01:00