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