75 Commits

Author	SHA1	Message	Date
henryruhs	1dfd230fc5	Revert the config dicts	2025-03-11 14:43:10 +01:00
henryruhs	6bc44ad3d8	Fix CI	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	dcc5ccccd7	Extend Unet with more layers	2025-03-11 14:43:10 +01:00
henryruhs	5056b8df75	Variable AAD layer according to output size	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	cadbe9cf76	Follow the concept of layers and sequences	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	94480e16eb	Rename to AAD as this is not a full AIENet	2025-03-11 14:43:10 +01:00
henryruhs	bbcb1c35f0	Modernize AIENet	2025-03-11 14:43:10 +01:00
henryruhs	ee3fc40e83	Move output channels to config	2025-03-11 14:43:10 +01:00
henryruhs	14bbece850	Rename id_ to identity_	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	f482d46798	Remove VisionTensor from Discriminator	2025-03-11 14:43:09 +01:00
henryruhs	a6e1405c70	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	10ce04ed58	Introduce feature flag for Unet	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