Commit Graph

51 Commits

Author SHA1 Message Date
henryruhs 5bba2a1c69 Remove the condition from reconstruction loss 2025-03-11 14:43:10 +01:00
henryruhs 94480e16eb Rename to AAD as this is not a full AIENet 2025-03-11 14:43:10 +01:00
henryruhs bbcb1c35f0 Modernize AIENet 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 fad38da864 Rename temp1 with positive and temp2 with negative 2025-03-11 14:43:10 +01:00
henryruhs e75a3c58f9 Replace calc() with forward(), Rename temp1 with positive and temp2 with negative 2025-03-11 14:43:10 +01:00
harisreedhar 6fed877d33 reconstruction-loss fix 2025-03-11 14:43:10 +01:00
henryruhs ed0f6ae897 Use new loss code, Remove unused code, Remove old types, Ban VisionTensor naming 2025-03-11 14:43:10 +01:00
henryruhs 63e4bea3cd Use new config for old code 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 579d3ef51c Introduce new GazeLoss class (switched to mean) 2025-03-11 14:43:10 +01:00
henryruhs a797548329 Introduce new PoseLoss class (switched to mean) 2025-03-11 14:43:10 +01:00
henryruhs 6eabcad1d0 Introduce new AttributeLoss class 2025-03-11 14:43:10 +01:00
henryruhs 7848d28b02 Introduce new AdversarialLoss class fix 2025-03-11 14:43:10 +01:00
henryruhs 30e787129a Introduce new AdversarialLoss class 2025-03-11 14:43:09 +01:00
henryruhs f2833a32c3 Introduce new AdversarialLoss class 2025-03-11 14:43:09 +01:00
henryruhs 3b7d3b6688 Introduce new ReconstructionLoss class 2025-03-11 14:43:09 +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 f482d46798 Remove VisionTensor from Discriminator 2025-03-11 14:43:09 +01:00
henryruhs a6e1405c70 Restore map_location = 'cpu' 2025-03-11 14:43:09 +01:00
henryruhs 206a1411d1 Fix Generator 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 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
harisreedhar 4d8433f54a changes 2025-03-11 14:43:09 +01:00
harisreedhar db44c91dd8 changes 2025-03-11 14:43:09 +01:00
henryruhs c161da2f25 Adjust naming and typing 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
harisreedhar 030d912c1b some fixes 2025-03-11 14:43:09 +01:00
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