Refactor env queue, Training diffusion works (Still not converging)

2024-03-04 10:59:43 +00:00
parent fddd9f0311
commit cfc304e870
11 changed files with 96 additions and 111 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:
@@ -22,6 +22,7 @@ save_buffer: false
 train_steps: ???
 fps: ???

+n_action_steps: ???
 env: ???

 policy: ???