Merge pull request #6 from Cadene/user/rcadene/2024_03_04_diffusion

Make diffusion work
2024-03-04 18:30:40 +01:00
parent d3aae3111c e29fbb50e8
commit e990f3e148
12 changed files with 276 additions and 142 deletions
--- a/lerobot/configs/default.yaml
+++ b/lerobot/configs/default.yaml
@@ -1,7 +1,7 @@
 defaults:
  - _self_
-  - env: simxarm
-  - policy: tdmpc
+  - env: pusht
+  - policy: diffusion

 hydra:
  run:
@@ -21,6 +21,7 @@ save_buffer: false
 train_steps: ???
 fps: ???

+n_action_steps: ???
 env: ???

 policy: ???