novastar112/pusht_96_norm4_10hz_remap
pusht_96_norm4_10hz 96px PushT PPO successful trajectory dataset. The trajectories are generated by a 10Hz PPO PushT solver with action codec norm4, rendered images in history[*].image, image_prev, and image_next at 96x96 and JPEG quality 85. Each episode is success-only and capped at 150 environment steps. Splits split records format train 500,000 gzip-compressed JSONL test 1,000 gzip-compressed JSONL Train/test initial states are filtered to be… See the full description on the dataset page: https://huggingface.co/datasets/novastar112/pusht_96_norm4_10hz_remap.
pusht96norm4_10hz
96px PushT PPO successful trajectory dataset.
The trajectories are generated by a 10Hz PPO PushT solver with action codec norm4, rendered images in history[*].image, image_prev, and image_next at 96x96 and JPEG quality 85.
Each episode is success-only and capped at 150 environment steps.
Splits
Train/test initial states are filtered to be disjoint by init_state_hash; see metadata/.
Coordinates in move actions use normalized [0, 1] coordinates rounded to exactly four decimal places; execution decodes them back to PushT 512x512 physical coordinates. Rendered images are 96px.
Target Hub repo: https://huggingface.co/datasets/novastar112/pusht_96_norm4_10hz
Remap
This dataset is remapped from novastar112/pusht_96_norm4_10hz. Each history step has image_prev rewritten to the same frame as image; the last history step of each episode also has image_next rewritten to that final image. Records are batched into fewer .jsonl.gz shards for faster snapshot downloads.
Remapped Hub repo: https://huggingface.co/datasets/novastar112/pusht_96_norm4_10hz_remap