140 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
henryruhs	ac41bab3a2	Restore map_location = 'cpu'	2025-03-11 14:43:09 +01:00
henryruhs	206a1411d1	Fix Generator	2025-03-11 14:43:09 +01:00
henryruhs	dc0abff0ce	Minor typo	2025-03-11 14:43:09 +01:00
henryruhs	10ce04ed58	Introduce feature flag for Unet	2025-03-11 14:43:09 +01:00
henryruhs	094d5cea9e	Fix CI	2025-03-11 14:43:09 +01:00
henryruhs	f6c59257d9	Partial use Resnet34 as a DownSample replacement	2025-03-11 14:43:09 +01:00
henryruhs	83ef075b1d	Remove map_location	2025-03-11 14:43:09 +01:00
henryruhs	575f215408	lr only in optimizer	2025-03-11 14:43:09 +01:00
henryruhs	d153c68813	Better split_ratio	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	84503761b9	follow the not invented here syndrome	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	3d9ff4add0	Adjust validation to one time per epoch	2025-03-11 14:43:09 +01:00
henryruhs	5934b47961	Add cosine_similarity	2025-03-11 14:43:09 +01:00
henryruhs	04eaa831ea	Rename to resume_path only	2025-03-11 14:43:09 +01:00
henryruhs	c1bed34c27	More tweaks	2025-03-11 14:43:09 +01:00
henryruhs	4078681031	Fix typing	2025-03-11 14:43:09 +01:00
henryruhs	15ee6fa763	Simplify sizes	2025-03-11 14:43:09 +01:00
henryruhs	251e610f0e	Follow the lightning naming and call this dataset, Improve config and types	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	c041073953	Modernize data loader, remove read image helper	2025-03-11 14:43:09 +01:00
henryruhs	354315502b	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
harisreedhar	d1bf54276d	changes	2025-03-11 14:43:09 +01:00
harisreedhar	b47c6b72ee	changes	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	ccf6fa7f43	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	f63bc788ac	Fix typo	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