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
|
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
|
6eabcad1d0
|
Introduce new AttributeLoss class
|
2025-03-11 14:43:10 +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
|
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
|
c17378f3c7
|
Let's call it batch ratio
|
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 |
|
henryruhs
|
4078681031
|
Fix typing
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
15ee6fa763
|
Simplify sizes
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
251e610f0e
|
Follow the lightning naming and call this dataset, Improve config and types
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
da51c5336d
|
This can be marked static
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
c041073953
|
Modernize data loader, remove read image helper
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
354315502b
|
Modernize data loader, remove read image helper
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
254bc17c98
|
Modernize data loader, remove read image helper
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
3cf9711df0
|
Modernize data loader, remove read image helper
|
2025-03-11 14:43:09 +01:00 |
|
harisreedhar
|
d1bf54276d
|
changes
|
2025-03-11 14:43:09 +01:00 |
|
harisreedhar
|
b47c6b72ee
|
changes
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
0d45568bd1
|
Uniform validation, Add cosine_similarity validation to face swapper
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
ccf6fa7f43
|
Uniform validation, Add cosine_similarity validation to face swapper
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
28977d37d6
|
Uniform validation, Add cosine_similarity validation to face swapper
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
11bb9065ba
|
Uniform resume checkpoint approach
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
999f2c9cbe
|
Rename variable of EmbeddingConverterTrainer
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
777a8384c2
|
Extend README with TensorBoard guide
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
0949c1358b
|
Fix CI
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
d9e10a9f7c
|
Use lightning over pytorch_lightning import, Configure tensorboard
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
0b7d25a36e
|
Adjust licenses
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
026bcf0c97
|
Adjust licenses
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
e1e0c11bb5
|
Fix CI
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
650551c19b
|
Simplify Batch type
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
257ab668ee
|
Name what it is
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
62d897f9d8
|
Direct return layers
|
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
|
b6b4f9f65b
|
Improve lot of types, imports and names
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
e33bc0d52a
|
Modernize to use ModuleList, Fix some types
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
f3409c5ade
|
Modernize to use ModuleList, Fix some types
|
2025-03-11 14:43:09 +01:00 |
|