Commit Graph

16 Commits

Author SHA1 Message Date
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