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

<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