5 Commits

Author	SHA1	Message	Date
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