harisreedhar
|
35b779b1ed
|
change pretrained models mode to eval
|
2025-03-11 14:43:10 +01:00 |
|
harisreedhar
|
ea1b0205f0
|
constructor injection
|
2025-03-11 14:43:10 +01:00 |
|
harisreedhar
|
a5eb7d6aa1
|
change Adam to AdamW
|
2025-03-11 14:43:10 +01:00 |
|
harisreedhar
|
b27b8663e5
|
changes
|
2025-03-11 14:43:10 +01:00 |
|
harisreedhar
|
d87f6c0b15
|
changes
|
2025-03-11 14:43:10 +01:00 |
|
harisreedhar
|
5d1b90ff19
|
changes
|
2025-03-11 14:43:10 +01:00 |
|
harisreedhar
|
2ddcf52b66
|
changes
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
c8801ececd
|
Fix export using Trainer
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
84b4451366
|
Fix export using Trainer
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
cadbe9cf76
|
Follow the concept of layers and sequences
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
58a85a80bb
|
Use high float32 matmul precision
|
2025-03-11 14:43:10 +01:00 |
|
harisreedhar
|
ab0a59fb74
|
changes
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
578b07a7f4
|
Add StatefulDataloader, Manual trigger scheduler
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
7ce9d27097
|
Different naming
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
8c24c9ec27
|
Fix CI
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
0ad2556c4c
|
Improve optimizer configs
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
484a49c27d
|
Restore dataset behaviour for same person
|
2025-03-11 14:43:10 +01:00 |
|
harisreedhar
|
e8cc2bfff1
|
add gaze loss
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
a951d700fc
|
Restore RGB normalization
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
6eff69a41a
|
There is no need to make directories
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
5b3b2abdd7
|
Remove annoying Tuner, Match both trainer configs a bit more
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
bfcbd6bf95
|
Fix CI
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
607c55ff1f
|
Remove Numpy and CV2 to fully use Tensors
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
93cbbf52d0
|
Fix CI
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
f5cd6b6336
|
This should be a StaticDataset, Fix learning rate finder
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
257e5e56a4
|
Rename split ratio
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
f19908ccd6
|
Fix spacing for lambda
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
18a2531b54
|
Fix naming in dataset.py
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
84be7d1ffb
|
Remove deprecated argument
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
8b53c76a0a
|
Remove deprecated argument
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
7d8cb146a4
|
Use latest numpy
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
303cbfa024
|
Final refactoring for AAD done
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
bc174186eb
|
Final refactoring for AAD done
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
d7158749c2
|
Improve naming
|
2025-03-11 14:43:10 +01:00 |
|
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 |
|