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/ogb.yaml
+++ b/config/train/data/ogb.yaml
@@ -0,0 +1,13 @@
+dataset:
+  name: ogbench/cube_single_expert
+  num_steps: ${eval:'${wm.num_preds} + ${wm.history_size}'}
+  frameskip: 5
+  keys_to_load:
+    - pixels
+    - action
+    - observation
+  keys_to_cache:
+    - action
+    - observation
+  keys_to_merge:
+    proprio: proprio