Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
---
|
4 |
+
:hatched_chick: **pi0 pretrained model is available** :hatched_chick:
|
5 |
+
|
6 |
+
You can download and use this model with:
|
7 |
+
```python
|
8 |
+
policy = Pi0Policy.from_pretrained("lerobot/pi0")
|
9 |
+
action = policy.select_action(batch)
|
10 |
+
```
|
11 |
+
|
12 |
+
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):
|
13 |
+
```python
|
14 |
+
python lerobot/scripts/train.py \
|
15 |
+
--policy.path=lerobot/pi0 \
|
16 |
+
--dataset.repo_id=danaaubakirova/koch_test
|
17 |
+
```
|
18 |
+
|
19 |
+
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).
|