Commit Graph

18 Commits

Author SHA1 Message Date
henryruhs a971506271 Revert changes in generator 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 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 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
henryruhs 32dfdcf1b3 Remove unsued 2025-03-11 14:43:09 +01:00
henryruhs 494b84aecb Refactor discriminator to use ModuleList, Reduce complexity of layer creation 2025-03-11 14:43:09 +01:00
henryruhs 860771e482 Fix weight identity 2025-03-11 14:43:09 +01:00
henryruhs 4d2038d4ce Rename loss_id to loss_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 611618e413 Make Embedding great again 2025-03-11 14:43:03 +01:00
harisreedhar 6381e755d7 list join 2025-03-11 14:43:03 +01:00
harisreedhar 2ed558a873 cleanup 2025-03-11 14:43:03 +01:00
henryruhs b7e2d3ccd7 It should be models and networks 2025-03-11 14:43:03 +01:00