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
|
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
|
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
|
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
|
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
|
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 |
|
harisreedhar
|
4d8433f54a
|
changes
|
2025-03-11 14:43:09 +01:00 |
|
harisreedhar
|
db44c91dd8
|
changes
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
c161da2f25
|
Adjust naming and typing
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
e1ba81f220
|
Improve generator namings, Flip args to source then target
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
bf696be097
|
Improve generator namings, Flip args to source then target
|
2025-03-11 14:43:09 +01:00 |
|
harisreedhar
|
030d912c1b
|
some fixes
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
a971506271
|
Revert changes in generator
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
1872f99584
|
Refacto UNet
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
29e82f909a
|
Move nld to networks
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
3be8368eaa
|
Improve lot of types, imports and names
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
b6b4f9f65b
|
Improve lot of types, imports and names
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
34d0bc10ed
|
Fix append
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
b785525f3b
|
Fix append
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
dd320ea5be
|
Fix self
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
71e0ae34c0
|
Fix self
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
32dfdcf1b3
|
Remove unsued
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
494b84aecb
|
Refactor discriminator to use ModuleList, Reduce complexity of layer creation
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
860771e482
|
Fix weight identity
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
4d2038d4ce
|
Rename loss_id to loss_identity
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
1b6e7a6ca5
|
Rename ArcFace Converter to Embedding Converter, Add EmbeddingDataset, Add learning rate to config
|
2025-03-11 14:43:06 +01:00 |
|
henryruhs
|
611618e413
|
Make Embedding great again
|
2025-03-11 14:43:03 +01:00 |
|
harisreedhar
|
6381e755d7
|
list join
|
2025-03-11 14:43:03 +01:00 |
|
harisreedhar
|
2ed558a873
|
cleanup
|
2025-03-11 14:43:03 +01:00 |
|