readme
This commit is contained in:
parent
7132790faa
commit
f1cd0d83d2
|
@ -25,12 +25,11 @@ Designed by Physical Intelligence. Ported from Jax by Hugging Face.
|
||||||
Example of finetuning the pi0+FAST pretrained model (`pi0_fast_base` in `openpi`):
|
Example of finetuning the pi0+FAST pretrained model (`pi0_fast_base` in `openpi`):
|
||||||
```bash
|
```bash
|
||||||
python lerobot/scripts/train.py \
|
python lerobot/scripts/train.py \
|
||||||
--policy.path=lerobot/pi0fast \
|
--policy.path=lerobot/pi0fast_base \
|
||||||
--dataset.repo_id=danaaubakirova/koch_test
|
--dataset.repo_id=danaaubakirova/koch_test
|
||||||
```
|
```
|
||||||
|
|
||||||
Example of finetuning the pi0+FAST neural network with PaliGemma and expert Gemma
|
Example of training the pi0+FAST neural network with from scratch:
|
||||||
pretrained with VLM default parameters before pi0+FAST finetuning:
|
|
||||||
```bash
|
```bash
|
||||||
python lerobot/scripts/train.py \
|
python lerobot/scripts/train.py \
|
||||||
--policy.type=pi0fast \
|
--policy.type=pi0fast \
|
||||||
|
@ -39,7 +38,7 @@ python lerobot/scripts/train.py \
|
||||||
|
|
||||||
Example of using the pi0 pretrained model outside LeRobot training framework:
|
Example of using the pi0 pretrained model outside LeRobot training framework:
|
||||||
```python
|
```python
|
||||||
policy = PI0FASTPolicy.from_pretrained("lerobot/pi0fast")
|
policy = PI0FASTPolicy.from_pretrained("lerobot/pi0fast_base")
|
||||||
```
|
```
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
|
Loading…
Reference in New Issue