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
|
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
|
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 |
|
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
|
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
|
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
|
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
|
723e9fde78
|
follow the not invented here syndrome
|
2025-03-11 14:43:09 +01:00 |
|