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
parent 3dc14b5576
commit ece89730e6
8 changed files with 222 additions and 111 deletions
--- a/lerobot/configs/pusht.yaml
+++ b/lerobot/configs/pusht.yaml
@@ -5,8 +5,12 @@ hydra:
  job:
    name: pusht

+video_dir: tmp/2024_02_21_pusht
+
 # env
 env: pusht
 image_size: 96
 frame_skip: 1
-
+state_dim: 2
+action_dim: 2
+fps: 10