This dataset contains 10M transitions for the cube-triple-play task from OGBench. It was used for the experiments in the paper Trust Region Q Adjoint Matching.
Project Page | Code | Paper
Dataset Structure
10 files of ~1M transitions each (split for easier loading)
Each file includes:
Training set: 1,000 episodes × 1,001 steps
Validation set: 100 episodes × 1,001 steps
Observation dim: 46, Action dim: 5
Files… See the full description on the dataset page: https://huggingface.co/datasets/yonghoon96/cube-triple-play-10m-v0.