49 Commits

Author	SHA1	Message	Date
henryruhs	1dfd230fc5	Revert the config dicts	2025-03-11 14:43:10 +01:00
henryruhs	b829d5e42c	Migrate most to self.config and self.context	2025-03-11 14:43:10 +01:00
henryruhs	ab3b699124	Rework on config	2025-03-11 14:43:10 +01:00
harisreedhar	a89e51c2f8	remove grad for inference only models	2025-03-11 14:43:10 +01:00
harisreedhar	61f48d9246	changes	2025-03-11 14:43:10 +01:00
henryruhs	e9ea9cd9e5	Clean GazeLoss	2025-03-11 14:43:10 +01:00
harisreedhar	7cf5609c1f	changes	2025-03-11 14:43:10 +01:00
henryruhs	f9de4ce78a	Remove UnetPro, make float values visible in README	2025-03-11 14:43:10 +01:00
henryruhs	430c71d031	Adjust config and namings	2025-03-11 14:43:10 +01:00
harisreedhar	176dced1f6	changes	2025-03-11 14:43:10 +01:00
harisreedhar	df5895e266	changes	2025-03-11 14:43:10 +01:00
harisreedhar	f2d3f8a19f	changes	2025-03-11 14:43:10 +01:00
harisreedhar	dfd9e99aed	changes	2025-03-11 14:43:10 +01:00
henryruhs	a22adaf51f	Adjust some namings	2025-03-11 14:43:10 +01:00
henryruhs	0055c0c97f	Adjust some namings	2025-03-11 14:43:10 +01:00
harisreedhar	ea1b0205f0	constructor injection	2025-03-11 14:43:10 +01:00
harisreedhar	d87f6c0b15	changes	2025-03-11 14:43:10 +01:00
harisreedhar	5d1b90ff19	changes	2025-03-11 14:43:10 +01:00
harisreedhar	2ddcf52b66	changes	2025-03-11 14:43:10 +01:00
henryruhs	7ce9d27097	Different naming	2025-03-11 14:43:10 +01:00
henryruhs	5bba2a1c69	Remove the condition from reconstruction loss	2025-03-11 14:43:10 +01:00
henryruhs	fad38da864	Rename temp1 with positive and temp2 with negative	2025-03-11 14:43:10 +01:00
henryruhs	e75a3c58f9	Replace calc() with forward(), Rename temp1 with positive and temp2 with negative	2025-03-11 14:43:10 +01:00
harisreedhar	6fed877d33	reconstruction-loss fix	2025-03-11 14:43:10 +01:00
henryruhs	ed0f6ae897	Use new loss code, Remove unused code, Remove old types, Ban VisionTensor naming	2025-03-11 14:43:10 +01:00
henryruhs	63e4bea3cd	Use new config for old code	2025-03-11 14:43:10 +01:00
henryruhs	14b9bccafe	Introduce new DiscriminatorLoss class, Remove useless super call params	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	a797548329	Introduce new PoseLoss 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	7848d28b02	Introduce new AdversarialLoss class fix	2025-03-11 14:43:10 +01:00
henryruhs	30e787129a	Introduce new AdversarialLoss class	2025-03-11 14:43:09 +01:00
henryruhs	f2833a32c3	Introduce new AdversarialLoss class	2025-03-11 14:43:09 +01:00
henryruhs	3b7d3b6688	Introduce new ReconstructionLoss class	2025-03-11 14:43:09 +01:00
henryruhs	086d9eed87	Introduce new ReconstructionLoss class	2025-03-11 14:43:09 +01:00
henryruhs	085c493e18	Rename id_embedder to embedder, Tons of naming in training step, Introduce new IdentityLoss class	2025-03-11 14:43:09 +01:00
henryruhs	a6e1405c70	Restore map_location = 'cpu'	2025-03-11 14:43:09 +01:00
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
harisreedhar	4d8433f54a	changes	2025-03-11 14:43:09 +01:00
harisreedhar	db44c91dd8	changes	2025-03-11 14:43:09 +01:00
harisreedhar	030d912c1b	some fixes	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	860771e482	Fix weight identity	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
harisreedhar	2ed558a873	cleanup	2025-03-11 14:43:03 +01:00