PWM / PWM_code /RISE_aux /docs /data_preparation.md
hchen657Naza's picture
Add files using upload-large-folder tool
23e1f91 verified
|
Raw History Blame Contribute Delete
650 Bytes
# Data Preparation
The training code expects LeRobot-style data.
## Expected Dataset Layout
```text
<dataset_name>/
β”œβ”€β”€ data/chunk-000/episode_*.parquet
β”œβ”€β”€ meta/
β”‚ β”œβ”€β”€ info.json
β”‚ β”œβ”€β”€ episodes.jsonl
β”‚ β”œβ”€β”€ episodes_stats.jsonl
β”‚ └── tasks.jsonl
└── videos/chunk-000/*.mp4
```
## Useful Utilities
- ALOHA conversion example:
- `policy_and_value/policy_offline_and_value/examples/aloha_real/convert_to_lerobot.py`
- Lightweight LeRobot tools:
- `policy_and_value/policy_offline_and_value/mini_lerobot/`
- Dynamics-model video resize helper:
- `dynamics/dynamics_model/preprocess.sh`