File size: 748 Bytes
9ce4044 78cbf98 9ce4044 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
---
license: apache-2.0
---
**pi0 pretrained model is available**
You can download and use this model with:
```python
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](https://huggingface.co/spaces/lerobot/visualize_dataset?dataset=danaaubakirova%2Fkoch_test&episode=0):
```python
python lerobot/scripts/train.py \
--policy.path=lerobot/pi0 \
--dataset.repo_id=danaaubakirova/koch_test
```
Take a look at the [code](https://github.com/huggingface/lerobot/blob/main/lerobot/common/policies/pi0/modeling_pi0.py), [twitt](https://x.com/RemiCadene/status/1886824456674332979), [blog post](https://huggingface.co/blog/pi0). |