43 Commits

Author	SHA1	Message	Date
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
henryruhs	09e913233b	Follow convention of the other project	2025-03-11 14:43:09 +01:00
henryruhs	b4bbd862e2	Follow convention of the other project	2025-03-11 14:43:09 +01:00
harisreedhar	4d8433f54a	changes	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	9bd68c3d14	Similarity validation for embedding converter	2025-03-11 14:43:09 +01:00
henryruhs	0a50e2d706	Rename data loader	2025-03-11 14:43:09 +01:00
henryruhs	dcf19634d1	Brain fart	2025-03-11 14:43:09 +01:00
henryruhs	857365770f	Make validation step more solid, failed on empty checksums	2025-03-11 14:43:09 +01:00
henryruhs	d25f2865a9	Normalize validation output	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	28977d37d6	Uniform validation, Add cosine_similarity validation to face swapper	2025-03-11 14:43:09 +01:00
henryruhs	c161da2f25	Adjust naming and typing	2025-03-11 14:43:09 +01:00
henryruhs	bf696be097	Improve generator namings, Flip args to source then target	2025-03-11 14:43:09 +01:00
henryruhs	11bb9065ba	Uniform resume checkpoint approach	2025-03-11 14:43:09 +01:00
harisreedhar	7b2b8f0f85	fix discriminator training	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
harisreedhar	030d912c1b	some fixes	2025-03-11 14:43:09 +01:00
henryruhs	29e82f909a	Move nld to networks	2025-03-11 14:43:09 +01:00
henryruhs	494b84aecb	Refactor discriminator to use ModuleList, Reduce complexity of layer creation	2025-03-11 14:43:09 +01:00
henryruhs	b42d2b06e7	Improve generate_preview	2025-03-11 14:43:09 +01:00
henryruhs	11c038cb81	Fix monitor	2025-03-11 14:43:09 +01:00
henryruhs	4d2038d4ce	Rename loss_id to loss_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	611618e413	Make Embedding great again	2025-03-11 14:43:03 +01:00
harisreedhar	2ed558a873	cleanup	2025-03-11 14:43:03 +01:00
henryruhs	989a81c751	Mixed bag of cleanups	2025-03-11 14:43:03 +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	e45f46d355	cleaning	2025-03-11 14:43:03 +01:00
harisreedhar	008a221f55	cleaning	2025-03-11 14:43:03 +01:00
henryruhs	10826558b4	Fix indent to tabs	2025-03-11 14:43:03 +01:00
henryruhs	23ac63d55b	Fix more typing	2025-03-11 14:43:03 +01:00
harisreedhar	9e1c71b498	clean	2025-03-11 14:43:03 +01:00
harisreedhar	a1fd382659	ugly training code	2025-03-11 14:43:03 +01:00
harisreedhar	650268c06b	ugly training code	2025-03-11 14:43:03 +01:00
harisreedhar	fe28b6fffe	ugly training code	2025-03-11 14:43:02 +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