Add policies/factory, Add test, Add _self_ in config

2024-02-25 10:50:23 +00:00
parent 64b5920e94
commit 598bb496b0
13 changed files with 61 additions and 38 deletions
--- a/lerobot/configs/default.yaml
+++ b/lerobot/configs/default.yaml
@@ -40,6 +40,7 @@ state_dim: 4
 action_dim: 4

 # TDMPC
+policy: tdmpc

 # planning
 mpc: true
--- a/lerobot/configs/pusht.yaml
+++ b/lerobot/configs/pusht.yaml
@@ -1,4 +1,5 @@
 defaults:
+  - _self_
  - default

 hydra: