21 Commits

Author	SHA1	Message	Date
Alexander Soare	dc745e3037	Remove unused part of diffusion policy config	2024-03-27 13:05:13 +00:00
Alexander Soare	acf1174447	ready for review	2024-03-21 10:18:50 +00:00
Alexander Soare	d323993569	backup wip	2024-03-20 15:01:27 +00:00
Alexander Soare	32e3f71dd1	backup wip	2024-03-20 09:49:16 +00:00
Alexander Soare	896a11f60e	backup wip	2024-03-19 18:50:04 +00:00
Alexander Soare	ea17f4ce50	backup wip	2024-03-19 16:02:09 +00:00
Alexander Soare	88347965c2	revert dp changes, make act and tdmpc batch friendly	2024-03-18 19:18:21 +00:00
Alexander Soare	98484ac68e	ready for review	2024-03-12 21:59:01 +00:00
Alexander Soare	87fcc536f9	wip - still need to verify full training run	2024-03-11 18:45:21 +00:00
Alexander Soare	2a01487494	early training loss as expected	2024-03-11 13:34:04 +00:00
Simon Alibert	6c867d78ef	Integrate pusht env from diffusion	2024-03-10 16:33:03 +01:00
Remi Cadene	a027f4edfb	Add cfg.offline_prioritized_sampler	2024-03-04 23:08:52 +00:00
Remi Cadene	e29fbb50e8	Fix grad_clip_norm 0 -> 10, Fix normalization min_max to be per channel	2024-03-04 17:26:34 +00:00
Remi Cadene	cfc304e870	Refactor env queue, Training diffusion works (Still not converging)	2024-03-04 11:00:51 +00:00
Remi Cadene	0f2fa4d9ef	Add obs queue to pusht, Set n_obs_steps=2 for diffusion (Not fully tested)	2024-03-03 13:21:31 +00:00
Remi Cadene	661bda45ea	imagenet_norm: False	2024-03-02 17:18:58 +00:00
Cadene	0b9027f05e	Clean logging, Refactor	2024-02-29 23:21:27 +00:00
Cadene	ac90b9c3ee	Fix diffusion (rm transpose), Add prefetch	2024-02-28 17:45:01 +00:00
Cadene	cf5063e50e	Add diffusion policy (train and eval works, TODO: reproduce results)	2024-02-28 15:21:42 +00:00
Cadene	7df542445c	Small fix and improve logging message	2024-02-27 11:44:26 +00:00
Cadene	21670dce90	Refactor train, eval_policy, logger, Add diffusion.yaml (WIP)	2024-02-26 01:10:09 +00:00