83 Commits

Author	SHA1	Message	Date
henryruhs	83ef075b1d	Remove map_location	2025-03-11 14:43:09 +01:00
henryruhs	00dccf07b9	Fix tensor foobar	2025-03-11 14:43:09 +01:00
henryruhs	5a6e3393e2	Revert loss for the moment	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
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	b33281425a	This is not a TensorDataset	2025-03-11 14:43:09 +01:00
henryruhs	04eaa831ea	Rename to resume_path only	2025-03-11 14:43:09 +01:00
henryruhs	da51c5336d	This can be marked static	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	40dcef7fc7	Modernize data loader, remove read image helper	2025-03-11 14:43:09 +01:00
henryruhs	254bc17c98	Modernize data loader, remove read image helper	2025-03-11 14:43:09 +01:00
henryruhs	39c0313202	Modernize data loader, remove read image helper	2025-03-11 14:43:09 +01:00
henryruhs	3cf9711df0	Modernize data loader, remove read image helper	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	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