51 Commits

Author	SHA1	Message	Date
henryruhs	803902c8bb	Add sync_dist to validation score logging	2025-03-16 09:01:40 +01:00
henryruhs	20aa5114e1	Extend config	2025-03-12 16:24:03 +01:00
henryruhs	2af4f2c8ab	Extend config	2025-03-12 16:21:47 +01:00
henryruhs	8101b15e1c	Revert the config dicts	2025-03-11 14:43:10 +01:00
henryruhs	1dfd230fc5	Revert the config dicts	2025-03-11 14:43:10 +01:00
henryruhs	a5d99c139e	Fix bug	2025-03-11 14:43:10 +01:00
henryruhs	d215b6f98b	Fix CI	2025-03-11 14:43:10 +01:00
henryruhs	01278d679f	Forward config parser	2025-03-11 14:43:10 +01:00
henryruhs	f4a1e18ca9	More adjustments	2025-03-11 14:43:10 +01:00
henryruhs	7ab7efbbf4	More adjustments	2025-03-11 14:43:10 +01:00
henryruhs	6a5f81e5fe	More adjustments	2025-03-11 14:43:10 +01:00
henryruhs	5bacb048dd	More adjustments	2025-03-11 14:43:10 +01:00
henryruhs	847579f925	More adjustments	2025-03-11 14:43:10 +01:00
henryruhs	57aad5204e	More adjustments	2025-03-11 14:43:10 +01:00
henryruhs	d3b0051912	More adjustments	2025-03-11 14:43:10 +01:00
henryruhs	d944d95bfd	Fix CI	2025-03-11 14:43:10 +01:00
henryruhs	b59e172fa3	Always config injection	2025-03-11 14:43:10 +01:00
henryruhs	e61e470432	Fix CI	2025-03-11 14:43:10 +01:00
henryruhs	c8953ce8a1	Apply new config approach for embedding converter	2025-03-11 14:43:10 +01:00
henryruhs	56be3f0b9b	Rename to validation_score	2025-03-11 14:43:10 +01:00
henryruhs	58a85a80bb	Use high float32 matmul precision	2025-03-11 14:43:10 +01:00
henryruhs	578b07a7f4	Add StatefulDataloader, Manual trigger scheduler	2025-03-11 14:43:10 +01:00
henryruhs	0ad2556c4c	Improve optimizer configs	2025-03-11 14:43:10 +01:00
henryruhs	6eff69a41a	There is no need to make directories	2025-03-11 14:43:10 +01:00
henryruhs	5b3b2abdd7	Remove annoying Tuner, Match both trainer configs a bit more	2025-03-11 14:43:10 +01:00
henryruhs	f5cd6b6336	This should be a StaticDataset, Fix learning rate finder	2025-03-11 14:43:10 +01:00
henryruhs	257e5e56a4	Rename split ratio	2025-03-11 14:43:10 +01:00
henryruhs	579d3ef51c	Introduce new GazeLoss class (switched to mean)	2025-03-11 14:43:10 +01:00
henryruhs	6eabcad1d0	Introduce new AttributeLoss class	2025-03-11 14:43:10 +01:00
henryruhs	a6e1405c70	Restore map_location = 'cpu'	2025-03-11 14:43:09 +01:00
henryruhs	575f215408	lr only in optimizer	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
harisreedhar	d1bf54276d	changes	2025-03-11 14:43:09 +01:00
harisreedhar	b47c6b72ee	changes	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	11bb9065ba	Uniform resume checkpoint approach	2025-03-11 14:43:09 +01:00
henryruhs	999f2c9cbe	Rename variable of EmbeddingConverterTrainer	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
henryruhs	b6b4f9f65b	Improve lot of types, imports and names	2025-03-11 14:43:09 +01:00
henryruhs	3c6dfa4efe	Modernize to use ModuleList, Fix some types	2025-03-11 14:43:09 +01:00
henryruhs	746cc86d52	Fix monitor	2025-03-11 14:43:09 +01:00