Files
lewm/config/train/data/ogb.yaml
2026-03-23 14:22:26 +01:00

13 lines
264 B
YAML

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