Robotics
Safetensors
lerobot
pi0 / README.md
cadene's picture
cadene HF Staff
Update README.md
78cbf98 verified
|
raw
history blame
748 Bytes
metadata
license: apache-2.0

pi0 pretrained model is available

You can download and use this model with:

policy = Pi0Policy.from_pretrained("lerobot/pi0")
action = policy.select_action(batch)

You can easily finetune it on your dataset. For instance on @dana_55517 's dataset:

python lerobot/scripts/train.py \
--policy.path=lerobot/pi0 \
--dataset.repo_id=danaaubakirova/koch_test

Take a look at the code, twitt, blog post.