Commit Graph

167 Commits

Author SHA1 Message Date
henryruhs 8b2b6892aa Improve naming 2025-03-11 14:43:10 +01:00
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 335d597e53 Rename to aienet.py 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 de6cfbc35b Fix CI 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 38211f0340 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 2220f5ef08 Add missing encoder_type to config 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 ac41bab3a2 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 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 094d5cea9e Fix CI 2025-03-11 14:43:09 +01:00
henryruhs f6c59257d9 Partial use Resnet34 as a DownSample replacement 2025-03-11 14:43:09 +01:00
henryruhs 83ef075b1d Remove map_location 2025-03-11 14:43:09 +01:00
henryruhs 575f215408 lr only in optimizer 2025-03-11 14:43:09 +01:00
henryruhs d153c68813 Better split_ratio 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
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
henryruhs 723e9fde78 follow the not invented here syndrome 2025-03-11 14:43:09 +01:00
henryruhs 09e913233b Follow convention of the other project 2025-03-11 14:43:09 +01:00
henryruhs b4bbd862e2 Follow convention of the other project 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 b33281425a This is not a TensorDataset 2025-03-11 14:43:09 +01:00
henryruhs 3d9ff4add0 Adjust validation to one time per epoch 2025-03-11 14:43:09 +01:00
henryruhs 5934b47961 Add cosine_similarity 2025-03-11 14:43:09 +01:00
henryruhs 04eaa831ea Rename to resume_path only 2025-03-11 14:43:09 +01:00
henryruhs c1bed34c27 More tweaks 2025-03-11 14:43:09 +01:00