Fix unit tests, Refactor, Add pusht env, (TODO pusht replay buffer, image preprocessing)

2024-02-20 12:26:57 +00:00
parent fdfb2010fd
commit 3da6ffb2cb
10 changed files with 559 additions and 89 deletions
--- a/lerobot/configs/pusht.yaml
+++ b/lerobot/configs/pusht.yaml
@@ -0,0 +1,12 @@
+defaults:
+  - default
+
+hydra:
+  job:
+    name: pusht
+
+# env
+env: pusht
+image_size: 96
+frame_skip: 1
+