peterchenyipu/

cloth_dataset

Four splits of a cloth-being-pushed-over-a-sphere dataset, generated with examples/generate_cloth_sphere_push.py (see gendata.sh at repo root). The train/test split is defined by cloth size (particles per side):

Split Directory Episodes Cloth size range Seed

Train cloth_push_sphere/ 2000 100–150 × 100–150 42

ID test cloth_push_sphere_id_test/ 100 100–150 × 100–150 88

OOD test (smaller) cloth_push_sphere_ood_test_smaller/ 50 50–100 × 50–100… See the full description on the dataset page: https://huggingface.co/datasets/peterchenyipu/cloth_dataset.