Update README.md
Browse files
README.md
CHANGED
|
@@ -38,8 +38,8 @@ python src/lerobot/scripts/train.py \
|
|
| 38 |
--policy.type=pi05 \
|
| 39 |
--output_dir=./outputs/pi05_training \
|
| 40 |
--job_name=pi05_training \
|
| 41 |
-
--policy.repo_id=
|
| 42 |
-
--policy.pretrained_path=
|
| 43 |
--policy.compile_model=true \
|
| 44 |
--policy.gradient_checkpointing=true \
|
| 45 |
--wandb.enable=true \
|
|
@@ -50,18 +50,6 @@ python src/lerobot/scripts/train.py \
|
|
| 50 |
--batch_size=32
|
| 51 |
```
|
| 52 |
|
| 53 |
-
## Conversion Details
|
| 54 |
-
|
| 55 |
-
This model was converted from JAX to PyTorch using the OpenPI conversion script:
|
| 56 |
-
|
| 57 |
-
```bash
|
| 58 |
-
python examples/convert_jax_model_to_pytorch.py \
|
| 59 |
-
--checkpoint_dir /pi05_base \
|
| 60 |
-
--config_name pi05_base \
|
| 61 |
-
--output_path /pi05_base/pytorch/fp32/ \
|
| 62 |
-
--precision float32
|
| 63 |
-
```
|
| 64 |
-
|
| 65 |
## Citation
|
| 66 |
|
| 67 |
If you use this model, please cite the original OpenPI work:
|
|
|
|
| 38 |
--policy.type=pi05 \
|
| 39 |
--output_dir=./outputs/pi05_training \
|
| 40 |
--job_name=pi05_training \
|
| 41 |
+
--policy.repo_id=your_repo_id \
|
| 42 |
+
--policy.pretrained_path=lerobot/pi05_base \
|
| 43 |
--policy.compile_model=true \
|
| 44 |
--policy.gradient_checkpointing=true \
|
| 45 |
--wandb.enable=true \
|
|
|
|
| 50 |
--batch_size=32
|
| 51 |
```
|
| 52 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
## Citation
|
| 54 |
|
| 55 |
If you use this model, please cite the original OpenPI work:
|