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
|
2220f5ef08
|
Add missing encoder_type to config
|
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
|
dc0abff0ce
|
Minor typo
|
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
|
575f215408
|
lr only in optimizer
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
d153c68813
|
Better split_ratio
|
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
|
84503761b9
|
follow the not invented here syndrome
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
723e9fde78
|
follow the not invented here syndrome
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
09e913233b
|
Follow convention of the other project
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
b4bbd862e2
|
Follow convention of the other project
|
2025-03-11 14:43:09 +01:00 |
|
harisreedhar
|
4d8433f54a
|
changes
|
2025-03-11 14:43:09 +01:00 |
|
harisreedhar
|
db44c91dd8
|
changes
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
b33281425a
|
This is not a TensorDataset
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
3d9ff4add0
|
Adjust validation to one time per epoch
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
5934b47961
|
Add cosine_similarity
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
04eaa831ea
|
Rename to resume_path only
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
c1bed34c27
|
More tweaks
|
2025-03-11 14:43:09 +01:00 |
|