Datasetpublic

robomotic/causality-two-room-modal

robomotic/causality-two-room-modal This dataset contains trajectories collected in the swm/GlitchedHueTwoRoom-v1 environment for causal world-model experiments. What is inside glitched_hue_tworoom.h5: HDF5 dataset with trajectories and rendered frames. Includes observations, actions, rewards, episode indexing, variation values, and pixel renderings. How it was generated Collection script: scripts/data/collect_glitched_hue.py Configuration:… See the full description on the dataset page: https://huggingface.co/datasets/robomotic/causality-two-room-modal.

sourceHugging Facemitupdated 5mo agoView on Hugging Face

0likes71downloads

Dataset Card

robomotic/causality-two-room-modal

This dataset contains trajectories collected in the swm/GlitchedHueTwoRoom-v1 environment for causal world-model experiments.

What is inside

—glitched_hue_tworoom.h5: HDF5 dataset with trajectories and rendered frames.
—Includes observations, actions, rewards, episode indexing, variation values, and pixel renderings.

How it was generated

Collection script:

—scripts/data/collect_glitched_hue.py

Configuration:

—scripts/data/config/glitched_hue.yaml

Command used:

bash

python scripts/data/collect_glitched_hue.py num_traj=10000 seed=3072 world.num_envs=10

Generation details:

—Total episodes: 20000
—Total frames: 1225533
—Pixel shape: (1225533, 224, 224, 3)
—Teleport events: 3166
—File size: 18.18 GB
—Uploaded at (UTC): 2026-04-20 18:25:55

Repository

—GitHub: https://github.com/epokhcs/stable-worldmodel

Notes

The second background hue is green (not red) to preserve strong contrast with the red agent in pixel observations.