Robotics
Safetensors
lerobot
cadene HF staff commited on
Commit
9ce4044
·
verified ·
1 Parent(s): cde6322

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -0
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).