Commit Graph

254 Commits

Author SHA1 Message Date
henryruhs 866019d44f Debug path 2025-03-11 14:43:10 +01:00
henryruhs 94ad33cb1e Debug path 2025-03-11 14:43:10 +01:00
henryruhs de72e50233 Only ubuntu needed for test 2025-03-11 14:43:10 +01:00
henryruhs 5f053f9f69 Crazy fix 2025-03-11 14:43:10 +01:00
henryruhs f678aa8f7e Crazy fix 2025-03-11 14:43:10 +01:00
henryruhs 0e8207ccc8 Crazy fix 2025-03-11 14:43:10 +01:00
henryruhs 4a2559c866 Crazy fix 2025-03-11 14:43:10 +01:00
henryruhs 3c5554c1c5 Crazy fix 2025-03-11 14:43:10 +01:00
henryruhs dbf5687bcd Crazy fix 2025-03-11 14:43:10 +01:00
henryruhs 2148e9b701 Prepare test for 512 2025-03-11 14:43:10 +01:00
henryruhs 64ebfa7b84 Add basic test for aad and unet 2025-03-11 14:43:10 +01:00
henryruhs daeec46e36 Add basic test for aad and unet 2025-03-11 14:43:10 +01:00
henryruhs 13d15029b7 Add basic test for aad and unet 2025-03-11 14:43:10 +01:00
henryruhs d5c51a90e8 Add basic test for aad and unet 2025-03-11 14:43:10 +01:00
henryruhs 6fa8d6b6eb Add basic test for aad and unet 2025-03-11 14:43:10 +01:00
henryruhs 72371b9f11 Add basic test for aad and unet 2025-03-11 14:43:10 +01:00
henryruhs 786adf73a2 Fix UnetPro 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 430c71d031 Adjust config and namings 2025-03-11 14:43:10 +01:00
harisreedhar 176dced1f6 changes 2025-03-11 14:43:10 +01:00
henryruhs 18a605e1a3 Change the weight config order 2025-03-11 14:43:10 +01:00
henryruhs fea75ff949 Remove useless comma 2025-03-11 14:43:10 +01:00
harisreedhar df5895e266 changes 2025-03-11 14:43:10 +01:00
harisreedhar f2d3f8a19f changes 2025-03-11 14:43:10 +01:00
henryruhs ceb3c0cfdf Rename to AugmentTransform 2025-03-11 14:43:10 +01:00
henryruhs dd0a2fe649 Add albumentations 2025-03-11 14:43:10 +01:00
henryruhs e5a4a54e61 Introduce Transform classes, Add albumentations 2025-03-11 14:43:10 +01:00
henryruhs 9dc1031fa5 Fix same and equal 2025-03-11 14:43:10 +01:00
henryruhs d93daf9e5f Introduce batch mode via config for equal and same batches 2025-03-11 14:43:10 +01:00
henryruhs 3a61da8bab Introduce batch mode via config for equal and same batches 2025-03-11 14:43:10 +01:00
harisreedhar 83c20f8331 changes 2025-03-11 14:43:10 +01:00
harisreedhar dfd9e99aed changes 2025-03-11 14:43:10 +01:00
harisreedhar 2fb0b4289d changes 2025-03-11 14:43:10 +01:00
harisreedhar e6ea454360 supress warning 2025-03-11 14:43:10 +01:00
henryruhs 80d1694e23 Replace naming warp_matrix with warp_template 2025-03-11 14:43:10 +01:00
henryruhs 5e68de9170 Add config to dataloader alignment 2025-03-11 14:43:10 +01:00
henryruhs 6ca68f1408 Add config to dataloader alignment 2025-03-11 14:43:10 +01:00
henryruhs 43f99db5d7 Add config to dataloader alignment 2025-03-11 14:43:10 +01:00
henryruhs c9e70ebc18 Add config to dataloader alignment 2025-03-11 14:43:10 +01:00
henryruhs 1ab57b1d3f Add config to dataloader alignment 2025-03-11 14:43:10 +01:00
henryruhs cc9a0ba83e Add config to dataloader alignment 2025-03-11 14:43:10 +01:00
henryruhs 589568bfb5 Add config to dataloader alignment 2025-03-11 14:43:10 +01:00
harisreedhar 34a7f3ef55 changes 2025-03-11 14:43:10 +01:00
harisreedhar c35b0a6f4c changes 2025-03-11 14:43:10 +01:00
harisreedhar 23ab7dc89d fix 2025-03-11 14:43:10 +01:00
henryruhs 8443f3512a torch.float32 is the default 2025-03-11 14:43:10 +01:00
henryruhs d3a2035d7a Fix order 2025-03-11 14:43:10 +01:00
henryruhs 16c8b32269 Fix some naming and types 2025-03-11 14:43:10 +01:00
harisreedhar 65ab796835 changes 2025-03-11 14:43:10 +01:00