From f9af76ea502d404240cf9394dbda5c6a0e4ec1e5 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Tue, 16 Sep 2025 14:42:31 +0200 Subject: [PATCH] change docs: finetune base model options --- docs/source/pi0.mdx | 6 +++--- docs/source/pi05.mdx | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/docs/source/pi0.mdx b/docs/source/pi0.mdx index 31d6356fb..35247ce3a 100644 --- a/docs/source/pi0.mdx +++ b/docs/source/pi0.mdx @@ -103,9 +103,9 @@ python src/lerobot/scripts/train.py \ - **`--policy.dtype=bfloat16`**: Use mixed precision training for efficiency - **`--batch_size=32`**: Batch size for training, adapt this based on your GPU memory - **`--policy.pretrained_path=pepijn223/pi0_base_fp32`**: The base π₀ model you want to finetune, options are: - - \*\* [pepijn223/pi0_base_fp32](https://huggingface.co/pepijn223/pi0_base_fp32) - - \*\* [pepijn223/pi0_libero_fp32](https://huggingface.co/pepijn223/pi0_libero_fp32) (specifically trained on the Libero dataset) - - \*\* [pepijn223/pi0_droid_fp32](https://huggingface.co/pepijn223/pi0_droid_fp32) (specifically trained on the Droid dataset) + - [pepijn223/pi0_base_fp32](https://huggingface.co/pepijn223/pi0_base_fp32) + - [pepijn223/pi0_libero_fp32](https://huggingface.co/pepijn223/pi0_libero_fp32) (specifically trained on the Libero dataset) + - [pepijn223/pi0_droid_fp32](https://huggingface.co/pepijn223/pi0_droid_fp32) (specifically trained on the Droid dataset) ## License diff --git a/docs/source/pi05.mdx b/docs/source/pi05.mdx index c11ec7c1d..458858df5 100644 --- a/docs/source/pi05.mdx +++ b/docs/source/pi05.mdx @@ -106,9 +106,9 @@ python src/lerobot/scripts/train.py \ - **`--policy.dtype=bfloat16`**: Use mixed precision training for efficiency - **`--batch_size=32`**: Batch size for training, adapt this based on your GPU memory - **`--policy.pretrained_path=pepijn223/pi05_base_fp32`**: The base π₀.₅ model you want to finetune, options are: - - \*\* [pepijn223/pi05_base_fp32](https://huggingface.co/pepijn223/pi05_base_fp32) - - \*\* [pepijn223/pi05_libero_fp32](https://huggingface.co/pepijn223/pi05_libero_fp32) (specifically trained on the Libero dataset) - - \*\* [pepijn223/pi05_droid_fp32](https://huggingface.co/pepijn223/pi05_droid_fp32) (specifically trained on the Droid dataset) + - [pepijn223/pi05_base_fp32](https://huggingface.co/pepijn223/pi05_base_fp32) + - [pepijn223/pi05_libero_fp32](https://huggingface.co/pepijn223/pi05_libero_fp32) (specifically trained on the Libero dataset) + - [pepijn223/pi05_droid_fp32](https://huggingface.co/pepijn223/pi05_droid_fp32) (specifically trained on the Droid dataset) ## Performance Results