Commit Graph

31 Commits

Author SHA1 Message Date
henryruhs ad675ae633 Join MaskNet to guide generator 2025-03-16 12:23:08 +01:00
henryruhs eefc69a820 Fix CI 2025-03-16 08:41:31 +01:00
henryruhs 94571c5676 Rename attribute to feature 2025-03-16 08:39:35 +01:00
henryruhs 7c75b0d898 Fix attribute loss 2025-03-12 22:41:28 +01:00
henryruhs 0d73bcf918 Fix attribute loss 2025-03-12 22:37:06 +01:00
henryruhs 569df9e96d Let the generator return target attributes 2025-03-12 15:33:31 +01:00
henryruhs 0991745753 Remove Attributes 2025-03-11 19:17:26 +01:00
henryruhs 70ac772a34 Remove Attributes 2025-03-11 19:16:17 +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
henryruhs ab3b699124 Rework on config 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 dcc5ccccd7 Extend Unet with more layers 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 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 14b9bccafe Introduce new DiscriminatorLoss class, Remove useless super call params 2025-03-11 14:43:10 +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 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 b6b4f9f65b Improve lot of types, imports and names 2025-03-11 14:43:09 +01:00
henryruhs 611618e413 Make Embedding great again 2025-03-11 14:43:03 +01:00
harisreedhar 2ed558a873 cleanup 2025-03-11 14:43:03 +01:00