OGBench visual-cube-quadruple training set for SWM-Next world-model experiments.
600 episodes, balanced mixture: 300 expert / 150 noisy / 150 play (50% expert,
matching SWM's recipe). Collected with SWM's generation scripts, env seeds 0-299,
200 steps/episode, rendered 768x768, stored as JPEG-448.
path
contents
__.pt
one episode: frames (list of JPEG bytes), actions (T x 5 fp32), proprio
manifest.json
episode… See the full description on the dataset page:
https://huggingface.co/datasets/kevin510/swm-ogb-recipe-balanced.
