harisreedhar
|
7cf5609c1f
|
changes
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
f9de4ce78a
|
Remove UnetPro, make float values visible in README
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
5056b8df75
|
Variable AAD layer according to output size
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
430c71d031
|
Adjust config and namings
|
2025-03-11 14:43:10 +01:00 |
|
harisreedhar
|
176dced1f6
|
changes
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
18a605e1a3
|
Change the weight config order
|
2025-03-11 14:43:10 +01:00 |
|
harisreedhar
|
f2d3f8a19f
|
changes
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
3a61da8bab
|
Introduce batch mode via config for equal and same batches
|
2025-03-11 14:43:10 +01:00 |
|
harisreedhar
|
dfd9e99aed
|
changes
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
80d1694e23
|
Replace naming warp_matrix with warp_template
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
589568bfb5
|
Add config to dataloader alignment
|
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
|
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
|
14b9bccafe
|
Introduce new DiscriminatorLoss class, Remove useless super call params
|
2025-03-11 14:43:10 +01:00 |
|
henryruhs
|
2220f5ef08
|
Add missing encoder_type to config
|
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 |
|
harisreedhar
|
db44c91dd8
|
changes
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
04eaa831ea
|
Rename to resume_path only
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
11bb9065ba
|
Uniform resume checkpoint approach
|
2025-03-11 14:43:09 +01:00 |
|
harisreedhar
|
030d912c1b
|
some fixes
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
f05ff6cdb1
|
Enforce IR version for older onnxruntime
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
860771e482
|
Fix weight 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
|
9c15f584aa
|
Mixed bag of cleanups
|
2025-03-11 14:43:03 +01:00 |
|
harisreedhar
|
fcb3390796
|
again again cleaning
|
2025-03-11 14:43:03 +01:00 |
|
harisreedhar
|
3b8b6442fc
|
add infer and some cleaning
|
2025-03-11 14:43:03 +01:00 |
|
harisreedhar
|
cfcf0ee2bd
|
add glob image pattern to config
|
2025-03-11 14:43:03 +01:00 |
|
harisreedhar
|
d018dd0633
|
some cleaning and add exporting
|
2025-03-11 14:43:03 +01:00 |
|
harisreedhar
|
008a221f55
|
cleaning
|
2025-03-11 14:43:03 +01:00 |
|
harisreedhar
|
e6c2a64256
|
ugly training code
|
2025-03-11 14:43:02 +01:00 |
|
harisreedhar
|
8e53c6bc9f
|
new swapper
|
2025-03-11 14:43:02 +01:00 |
|