21 Commits

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