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 |
|
henryruhs
|
56be3f0b9b
|
Rename to validation_score
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
a22adaf51f
|
Adjust some namings
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
0055c0c97f
|
Adjust some namings
|
2025-03-11 14:43:10 +01:00 |
|
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 |
|