52 Commits

Author	SHA1	Message	Date
Cadene	ece89730e6	Add pusht dataset (TODO verify reward is aligned), Refactor visualize_dataset, Add video_dir, fps, state_dim, action_dim to config (Training works)	2024-02-21 00:49:40 +00:00
Cadene	3da6ffb2cb	Fix unit tests, Refactor, Add pusht env, (TODO pusht replay buffer, image preprocessing)	2024-02-20 12:26:57 +00:00