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
|
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
|
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 |
|
henryruhs
|
c6d16c0cf6
|
Modernize to use ModuleList, Fix some types
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
3c6dfa4efe
|
Modernize to use ModuleList, Fix some types
|
2025-03-11 14:43:09 +01:00 |
|
henryruhs
|
746cc86d52
|
Fix monitor
|
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 |
|