Commit Graph

6 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
Cadene
5a219fed6e Refactor policy config 2024-02-25 18:26:44 +00:00