qihuanye/lewm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Initial commit

Browse Source
This commit is contained in:
Lucas Maes
2026-03-12 22:56:21 -04:00
committed by lucas-maes
commit 83f97d72ad
21 changed files with 1355 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
config/train/data/dmc.yaml Normal file
View File

@@ -0,0 +1,11 @@
dataset:
num_steps: ${eval:'${wm.num_preds} + ${wm.history_size}'}
frameskip: 5
name: reacher
keys_to_load:
- pixels
- action
- observation
keys_to_cache:
- action
- observation

13
config/train/data/ogb.yaml Normal file
View File

@@ -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

13
config/train/data/pusht.yaml Normal file
View File

@@ -0,0 +1,13 @@
dataset:
num_steps: ${eval:'${wm.num_preds} + ${wm.history_size}'}
frameskip: 5
name: pusht_expert_train
keys_to_load:
- pixels
- action
- proprio
- state
keys_to_cache:
- action
- proprio
- state

11
config/train/data/tworoom.yaml Normal file
View File

@@ -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