5 Commits

Author	SHA1	Message	Date
Simon Alibert	7e024fdce6	Ran pre-commit run --all-files	2024-02-29 13:37:48 +01: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	21670dce90	Refactor train, eval_policy, logger, Add diffusion.yaml (WIP)	2024-02-26 01:10:09 +00:00
Cadene	b16c334825	Refactor configs to have env in seperate yaml + Fix training	2024-02-25 17:42:47 +00:00