Commit Graph

Select branches

Hide Pull Requests

master

#1

#10

#11

#12

#13

#14

#15

#16

#17

#18

#19

#2

#20

#21

#22

#23

#24

#25

#26

#27

#28

#29

#3

#30

#31

#32

#33

#34

#35

#36

#37

#38

#39

#4

#40

#41

#42

#43

#44

#45

#46

#47

#48

#49

#5

#50

#51

#52

#53

#54

#55

#56

#57

#58

#59

#6

#60

#61

#62

#63

#64

#65

#66

#67

#68

#69

#7

#70

#71

#72

#73

#74

#75

#76

#77

#78

#79

#8

#80

#81

#82

#83

#84

#85

#86

#87

#88

#89

#9

#90

#91

#92

#93

#94

#95

#96

models

Mono Color

• 7cc893c32e Adjust networks for 512 henryruhs 2025-03-05 15:23:31 +01:00
• 3e69d5a9a9 Adjust networks for 512 henryruhs 2025-03-05 15:23:16 +01:00
• f791178ded Fix CI henryruhs 2025-03-05 15:02:12 +01:00
• 866019d44f Debug path henryruhs 2025-03-05 14:59:59 +01:00
• 94ad33cb1e Debug path henryruhs 2025-03-05 14:50:50 +01:00
• de72e50233 Only ubuntu needed for test henryruhs 2025-03-05 14:44:58 +01:00
• 5f053f9f69 Crazy fix henryruhs 2025-03-05 14:42:21 +01:00
• f678aa8f7e Crazy fix henryruhs 2025-03-05 14:41:15 +01:00
• 0e8207ccc8 Crazy fix henryruhs 2025-03-05 14:36:07 +01:00
• 4a2559c866 Crazy fix henryruhs 2025-03-05 14:34:51 +01:00
• 3c5554c1c5 Crazy fix henryruhs 2025-03-05 14:32:27 +01:00
• dbf5687bcd Crazy fix henryruhs 2025-03-05 13:59:21 +01:00
• 2148e9b701 Prepare test for 512 henryruhs 2025-03-05 13:50:23 +01:00
• 64ebfa7b84 Add basic test for aad and unet henryruhs 2025-03-05 13:43:47 +01:00
• daeec46e36 Add basic test for aad and unet henryruhs 2025-03-05 13:28:08 +01:00
• 13d15029b7 Add basic test for aad and unet henryruhs 2025-03-05 13:26:44 +01:00
• d5c51a90e8 Add basic test for aad and unet henryruhs 2025-03-05 13:10:14 +01:00
• 6fa8d6b6eb Add basic test for aad and unet henryruhs 2025-03-05 12:57:15 +01:00
• 72371b9f11 Add basic test for aad and unet henryruhs 2025-03-05 12:53:24 +01:00
• 786adf73a2 Fix UnetPro henryruhs 2025-03-05 10:27:55 +01:00
• dcc5ccccd7 Extend Unet with more layers henryruhs 2025-03-05 10:25:37 +01:00
• 5056b8df75 Variable AAD layer according to output size henryruhs 2025-03-04 22:13:37 +01:00
• 430c71d031 Adjust config and namings henryruhs 2025-03-04 16:24:24 +01:00
• 176dced1f6 changes harisreedhar 2025-03-04 20:07:41 +05:30
• 18a605e1a3 Change the weight config order henryruhs 2025-03-04 12:19:49 +01:00
• fea75ff949 Remove useless comma henryruhs 2025-03-04 12:17:59 +01:00
• df5895e266 changes harisreedhar 2025-03-04 16:43:36 +05:30
• f2d3f8a19f changes harisreedhar 2025-03-04 14:23:20 +05:30
• ceb3c0cfdf Rename to AugmentTransform henryruhs 2025-03-03 18:29:51 +01:00
• dd0a2fe649 Add albumentations henryruhs 2025-03-03 18:24:09 +01:00
• e5a4a54e61 Introduce Transform classes, Add albumentations henryruhs 2025-03-03 18:23:18 +01:00
• 9dc1031fa5 Fix same and equal henryruhs 2025-03-03 12:21:47 +01:00
• d93daf9e5f Introduce batch mode via config for equal and same batches henryruhs 2025-03-03 12:15:01 +01:00
• 3a61da8bab Introduce batch mode via config for equal and same batches henryruhs 2025-03-03 12:13:54 +01:00
• 83c20f8331 changes harisreedhar 2025-03-03 16:42:21 +05:30
• dfd9e99aed changes harisreedhar 2025-03-03 16:41:26 +05:30
• 2fb0b4289d changes harisreedhar 2025-03-03 15:13:34 +05:30
• e6ea454360 supress warning harisreedhar 2025-03-03 15:12:35 +05:30
• 80d1694e23 Replace naming warp_matrix with warp_template henryruhs 2025-03-03 09:57:27 +01:00
• 5e68de9170 Add config to dataloader alignment henryruhs 2025-03-03 09:45:41 +01:00
• 6ca68f1408 Add config to dataloader alignment henryruhs 2025-03-03 09:45:02 +01:00
• 43f99db5d7 Add config to dataloader alignment henryruhs 2025-03-03 09:41:40 +01:00
• c9e70ebc18 Add config to dataloader alignment henryruhs 2025-03-03 09:41:04 +01:00
• 1ab57b1d3f Add config to dataloader alignment henryruhs 2025-03-03 09:39:56 +01:00
• cc9a0ba83e Add config to dataloader alignment henryruhs 2025-03-03 09:32:44 +01:00
• 589568bfb5 Add config to dataloader alignment henryruhs 2025-03-03 09:23:40 +01:00
• 34a7f3ef55 changes harisreedhar 2025-03-03 13:30:00 +05:30
• c35b0a6f4c changes harisreedhar 2025-03-03 13:28:55 +05:30
• 23ab7dc89d fix harisreedhar 2025-03-03 13:20:37 +05:30
• 8443f3512a torch.float32 is the default henryruhs 2025-03-03 08:46:20 +01:00
• d3a2035d7a Fix order henryruhs 2025-03-03 08:41:56 +01:00
• 16c8b32269 Fix some naming and types henryruhs 2025-03-03 08:33:52 +01:00
• 65ab796835 changes harisreedhar 2025-02-27 23:17:06 +05:30
• 56be3f0b9b Rename to validation_score henryruhs 2025-03-01 16:14:07 +01:00
• a22adaf51f Adjust some namings henryruhs 2025-03-01 16:06:38 +01:00
• 0055c0c97f Adjust some namings henryruhs 2025-03-01 16:02:35 +01:00
• 35b779b1ed change pretrained models mode to eval harisreedhar 2025-02-27 19:24:12 +05:30
• ea1b0205f0 constructor injection harisreedhar 2025-02-27 19:17:11 +05:30
• a5eb7d6aa1 change Adam to AdamW harisreedhar 2025-02-27 19:05:41 +05:30
• b27b8663e5 changes harisreedhar 2025-02-27 16:18:53 +05:30
• d87f6c0b15 changes harisreedhar 2025-02-27 15:46:50 +05:30
• 5d1b90ff19 changes harisreedhar 2025-02-27 13:13:20 +05:30
• 2ddcf52b66 changes harisreedhar 2025-02-27 12:18:09 +05:30
• c8801ececd Fix export using Trainer henryruhs 2025-02-26 16:56:02 +01:00
• 84b4451366 Fix export using Trainer henryruhs 2025-02-26 16:50:40 +01:00
• cadbe9cf76 Follow the concept of layers and sequences henryruhs 2025-02-26 14:19:26 +01:00
• 58a85a80bb Use high float32 matmul precision henryruhs 2025-02-26 10:54:38 +01:00
• ab0a59fb74 changes harisreedhar 2025-02-26 14:01:41 +05:30
• 578b07a7f4 Add StatefulDataloader, Manual trigger scheduler henryruhs 2025-02-26 00:43:42 +01:00
• 7ce9d27097 Different naming henryruhs 2025-02-26 00:03:57 +01:00
• 8c24c9ec27 Fix CI henryruhs 2025-02-26 00:02:54 +01:00
• 0ad2556c4c Improve optimizer configs henryruhs 2025-02-26 00:01:18 +01:00
• 484a49c27d Restore dataset behaviour for same person henryruhs 2025-02-25 12:54:54 +01:00
• e8cc2bfff1 add gaze loss harisreedhar 2025-02-25 13:59:02 +05:30
• a951d700fc Restore RGB normalization henryruhs 2025-02-24 16:21:37 +01:00
• 6eff69a41a There is no need to make directories henryruhs 2025-02-24 15:47:56 +01:00
• 5b3b2abdd7 Remove annoying Tuner, Match both trainer configs a bit more henryruhs 2025-02-24 15:40:50 +01:00
• bfcbd6bf95 Fix CI henryruhs 2025-02-24 12:24:56 +01:00
• 607c55ff1f Remove Numpy and CV2 to fully use Tensors henryruhs 2025-02-24 12:23:57 +01:00
• 93cbbf52d0 Fix CI henryruhs 2025-02-24 10:26:39 +01:00
• f5cd6b6336 This should be a StaticDataset, Fix learning rate finder henryruhs 2025-02-24 10:25:19 +01:00
• 257e5e56a4 Rename split ratio henryruhs 2025-02-24 09:39:54 +01:00
• f19908ccd6 Fix spacing for lambda henryruhs 2025-02-24 09:26:46 +01:00
• 18a2531b54 Fix naming in dataset.py henryruhs 2025-02-24 09:25:28 +01:00
• 84be7d1ffb Remove deprecated argument henryruhs 2025-02-24 00:49:38 +01:00
• 8b53c76a0a Remove deprecated argument henryruhs 2025-02-24 00:47:12 +01:00
• 7d8cb146a4 Use latest numpy henryruhs 2025-02-24 00:30:12 +01:00
• 303cbfa024 Final refactoring for AAD done henryruhs 2025-02-24 00:18:35 +01:00
• bc174186eb Final refactoring for AAD done henryruhs 2025-02-24 00:15:13 +01:00
• d7158749c2 Improve naming henryruhs 2025-02-23 22:07:30 +01:00
• 8b2b6892aa Improve naming henryruhs 2025-02-23 21:39:01 +01:00
• 5bba2a1c69 Remove the condition from reconstruction loss henryruhs 2025-02-23 19:10:38 +01:00
• 94480e16eb Rename to AAD as this is not a full AIENet henryruhs 2025-02-23 19:02:45 +01:00
• bbcb1c35f0 Modernize AIENet henryruhs 2025-02-23 17:21:12 +01:00
• ee3fc40e83 Move output channels to config henryruhs 2025-02-23 14:26:01 +01:00
• 335d597e53 Rename to aienet.py henryruhs 2025-02-23 14:13:37 +01:00
• 14bbece850 Rename id_ to identity_ henryruhs 2025-02-23 14:11:04 +01:00
• fad38da864 Rename temp1 with positive and temp2 with negative henryruhs 2025-02-23 09:35:18 +01:00
• e75a3c58f9 Replace calc() with forward(), Rename temp1 with positive and temp2 with negative henryruhs 2025-02-23 09:22:10 +01:00
• 6fed877d33 reconstruction-loss fix harisreedhar 2025-02-23 13:30:08 +05:30