65 Commits

Author	SHA1	Message	Date
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	39818a16df	Fix CI	2025-03-11 14:43:09 +01:00
henryruhs	e1ba81f220	Improve generator namings, Flip args to source then target	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	88c4e53192	Use skip_tensor variable	2025-03-11 14:43:09 +01:00
henryruhs	650551c19b	Simplify Batch type	2025-03-11 14:43:09 +01:00
henryruhs	a971506271	Revert changes in generator	2025-03-11 14:43:09 +01:00
henryruhs	b69f69d015	Fix UNet	2025-03-11 14:43:09 +01:00
henryruhs	953525e6b0	Fix typo	2025-03-11 14:43:09 +01:00
henryruhs	58ad6af619	Add Sample suffix again	2025-03-11 14:43:09 +01:00
henryruhs	7264884ff9	Fix CI	2025-03-11 14:43:09 +01:00
henryruhs	1872f99584	Refacto UNet	2025-03-11 14:43:09 +01:00
henryruhs	29e82f909a	Move nld to networks	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	3be8368eaa	Improve lot of types, imports and names	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	34d0bc10ed	Fix append	2025-03-11 14:43:09 +01:00
henryruhs	b785525f3b	Fix append	2025-03-11 14:43:09 +01:00
henryruhs	dd320ea5be	Fix self	2025-03-11 14:43:09 +01:00
henryruhs	71e0ae34c0	Fix self	2025-03-11 14:43:09 +01:00
henryruhs	32dfdcf1b3	Remove unsued	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	860771e482	Fix weight identity	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	58c81cd646	Move magic methods up	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	62a69cddd2	Fix import order	2025-03-11 14:43:03 +01:00
henryruhs	611618e413	Make Embedding great again	2025-03-11 14:43:03 +01:00
harisreedhar	6381e755d7	list join	2025-03-11 14:43:03 +01:00
harisreedhar	2ed558a873	cleanup	2025-03-11 14:43:03 +01:00
henryruhs	b7e2d3ccd7	It should be models and networks	2025-03-11 14:43:03 +01:00
henryruhs	a17d050648	Mixed bag of cleanups	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	cfcf0ee2bd	add glob image pattern to config	2025-03-11 14:43:03 +01:00
harisreedhar	4260bd6c28	fix type	2025-03-11 14:43:03 +01:00