Initial commit

2026-03-12 22:56:21 -04:00
commit 83f97d72ad
21 changed files with 1355 additions and 0 deletions
--- a/config/train/data/tworoom.yaml
+++ b/config/train/data/tworoom.yaml
@@ -0,0 +1,11 @@
+dataset:
+  num_steps: ${eval:'${wm.num_preds} + ${wm.history_size}'}
+  frameskip: 5
+  name: tworoom
+  keys_to_load:
+    - pixels
+    - action
+    - proprio
+  keys_to_cache:
+    - action
+    - proprio