Commit Graph

158 Commits

Author SHA1 Message Date
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
henryruhs 4078681031 Fix typing 2025-03-11 14:43:09 +01:00
henryruhs 15ee6fa763 Simplify sizes 2025-03-11 14:43:09 +01:00
henryruhs 251e610f0e Follow the lightning naming and call this dataset, Improve config and types 2025-03-11 14:43:09 +01:00
henryruhs da51c5336d This can be marked static 2025-03-11 14:43:09 +01:00
henryruhs 9bd68c3d14 Similarity validation for embedding converter 2025-03-11 14:43:09 +01:00
henryruhs 0a50e2d706 Rename data loader 2025-03-11 14:43:09 +01:00
henryruhs 40dcef7fc7 Modernize data loader, remove read image helper 2025-03-11 14:43:09 +01:00
henryruhs c041073953 Modernize data loader, remove read image helper 2025-03-11 14:43:09 +01:00
henryruhs 354315502b Modernize data loader, remove read image helper 2025-03-11 14:43:09 +01:00