Remove previous pi0 and rename pi0_openpi and pi05_openpi

This commit is contained in:
Pepijn
2025-09-22 17:11:29 +02:00
parent 83ed49d9b9
commit 5d9acf9d51
41 changed files with 54 additions and 2813 deletions
+4
View File
@@ -124,3 +124,7 @@ python src/lerobot/scripts/train.py \
LeRobot uses MuJoCo for simulation. You need to set the rendering backend before training or evaluation:
- `export MUJOCO_GL=egl` → for headless servers (e.g. HPC, cloud)
## Reproducing π₀ and π₀.₅ results
We can also reproduce the results of π₀ and π₀.₅ on the Libero benchmark by using the finetuned libero models.
+3 -3
View File
@@ -35,7 +35,7 @@ As described by Physical Intelligence, while AI has achieved remarkable success
2. Apply the custom patches:
```bash
cp -r ./src/lerobot/policies/pi0_openpi/transformers_replace/* \
cp -r ./src/lerobot/policies/pi0/transformers_replace/* \
$(python -c "import transformers, os; print(os.path.dirname(transformers.__file__))")
```
@@ -72,7 +72,7 @@ pip install transformers==4.53.2
To use π₀ in LeRobot, specify the policy type as:
```python
policy.type=pi0_openpi
policy.type=pi0
```
## Training
@@ -82,7 +82,7 @@ For training π₀, you can use the standard LeRobot training script with the ap
```bash
python src/lerobot/scripts/train.py \
--dataset.repo_id=your_dataset \
--policy.type=pi0_openpi \
--policy.type=pi0 \
--output_dir=./outputs/pi0_training \
--job_name=pi0_training \
--policy.pretrained_path=pepijn223/pi0_base_fp32 \
+3 -3
View File
@@ -43,7 +43,7 @@ This diverse training mixture creates a "curriculum" that enables generalization
2. Apply the custom patches:
```bash
cp -r ./src/lerobot/policies/pi05_openpi/transformers_replace/* \
cp -r ./src/lerobot/policies/pi05/transformers_replace/* \
$(python -c "import transformers, os; print(os.path.dirname(transformers.__file__))")
```
@@ -72,7 +72,7 @@ pip install transformers==4.53.2
To use π₀.₅ in your LeRobot configuration, specify the policy type as:
```python
policy.type=pi05_openpi
policy.type=pi05
```
## Training
@@ -84,7 +84,7 @@ Here's a complete training command for finetuning the base π₀.₅ model on yo
```bash
python src/lerobot/scripts/train.py \
--dataset.repo_id=your_dataset \
--policy.type=pi05_openpi \
--policy.type=pi05 \
--output_dir=./outputs/pi0_training \
--job_name=pi0_training \
--policy.repo_id=pepijn223/pi05_base_fp32 \