pi0-fast-ft-primitive-10task-deltachunk

Pi0-fast delta action chunk trained on VLABench datasets.

This repository provides the Pi0-fast model trained with the whole VLABench's official primitive tasks dataset. To be noticed, this config corresponds to the delta chunk, which is state[1:] - state[:-1] instead of state - state[0].

Evaluation

To run this checkpoint, please clone this repo: https://github.com/Shiduo-zh/openpi, and checkout to the branch main. Assume that you download this checkpoints and put it in the directory checkpoints, to run the policy as server, please run:

bash vla_bench_scipts/serve_policy.sh pi0_ft_vlabench_primitive_aligned checkpoints/VLABench/pi0-fast-ft-primitive-10task-deltachunk/29999/

After serving the policy, open another terminal and run:

bash vla_bench_scipts/multi_run_vlabench.sh <Your path to store the evaluate results>

Train

To reproduce the training result, please run the training script with the config pi0_ft_vlabench_primitive_aligned.

XLA_PYTHON_CLIENT_MEM_FRACTION=0.95 uv run scripts/train.py pi0_ft_vlabench_primitive_aligned --exp-name=pi0_ft_vlabench_primitive_aligned --overwrite

Our checkpoint is trained on 8 H100 for 30k iterations, with 5000 episodes data acrossing 10 tasks.

Reference Results

The reference success rate of this model is: