Skip to content

Commit 1232746

Browse files
author
Frédéric Legrand
committed
feat: add information about pi0.5
1 parent 8d0aa43 commit 1232746

File tree

2 files changed

+21
-21
lines changed

2 files changed

+21
-21
lines changed

mintlify/basic-usage/inference.mdx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ You just [trained an AI model](/basic-usage/training) and now you want to use it
1313

1414
## Control your robot with AI from the phosphobot dashboard
1515

16-
If you trained your model using phosphobot, you can control your robot directly from the phosphobot dashboard.
16+
If you trained your model using phosphobot, you can control your robot directly from the phosphobot dashboard.
1717

1818
<Note>
1919
You can fine tune the model in a single click from the dashboard. [Go here to
@@ -124,7 +124,7 @@ while True:
124124
title="Pi0"
125125
>
126126

127-
### Example script for Pi0
127+
### Example script for Pi0.5
128128

129129
```python
130130
#pip install --upgrade phosphobot

mintlify/learn/ai-models.mdx

Lines changed: 19 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,9 @@ title: "Train robotics AI models"
33
description: "A guide to training AI models that control robots"
44
---
55

6-
import InstallCode from '/snippets/install-code.mdx';
7-
import GetMQApp from '/snippets/get-mq-app.mdx';
8-
import TeleopInstructions from '/snippets/teleop-instructions.mdx';
9-
6+
import InstallCode from "/snippets/install-code.mdx";
7+
import GetMQApp from "/snippets/get-mq-app.mdx";
8+
import TeleopInstructions from "/snippets/teleop-instructions.mdx";
109

1110
The phospho starter pack makes it easy to train robotics AI models by integrating with **LeRobot** from Hugging Face.
1211

@@ -50,7 +49,7 @@ In this guide, we will use the phospho starter pack to record a dataset and uplo
5049
1. You need an assembled **SO-100** robot arm and **cameras**. Get the [phosphot starter pack here](https://robots.phospho.ai).
5150
2. Install [the phosphobot software](/installation)
5251

53-
<InstallCode/>
52+
<InstallCode />
5453

5554
3. **Connect your cameras to the computer.** Start the phosphobot server.
5655

@@ -61,7 +60,7 @@ phosphobot run
6160
4. Complete the [quickstart](/so-100/quickstart) and check that you can [control your robot](/basic-usage/teleop).
6261
5. You have the **[phosphobot teleoperation app](/examples/teleop)** is installed on your **Meta Quest 2, Pro, 3 or 3s**
6362

64-
<GetMQApp/>
63+
<GetMQApp />
6564

6665
6. You have a **device to train your model**. We recommend using a **GPU** for faster training.
6766

@@ -94,17 +93,18 @@ Go to the _Admin Configuration_ page of your phospshobot dashboard. You can adju
9493

9594
The easiest way to record a dataset is to use the Meta Quest app.
9695

97-
<GetMQApp/>
96+
<GetMQApp />
9897

99-
<iframe
100-
className="w-full aspect-video"
101-
src="https://www.youtube.com/embed/AQ-xgCTdj_w?si=NDTtX1nISABjzmbA"
102-
title="VR Control tutorial" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share"
103-
referrerPolicy="strict-origin-when-cross-origin"
104-
allowFullScreen>
105-
</iframe>
98+
<iframe
99+
className="w-full aspect-video"
100+
src="https://www.youtube.com/embed/AQ-xgCTdj_w?si=NDTtX1nISABjzmbA"
101+
title="VR Control tutorial"
102+
allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share"
103+
referrerPolicy="strict-origin-when-cross-origin"
104+
allowFullScreen
105+
></iframe>
106106

107-
<TeleopInstructions/>
107+
<TeleopInstructions />
108108

109109
Go to the **Dataset tab** in the phosphobot dashboard to see the recorded dataset. Use the button Preview to preview them using [LeRobot Dataset Visualizer](https://huggingface.co/spaces/lerobot/visualize_dataset).
110110

@@ -119,13 +119,14 @@ Go to the **Dataset tab** in the phosphobot dashboard to see the recorded datase
119119

120120
## 4. Train your first model
121121

122-
### Train GR00T-N1-2B in one click with phosphobot cloud
122+
### Train GR00T-N1-2B/Pi0.5/ACT/BB_ACT in one click with phosphobot cloud
123123

124124
To train a model, you can use the phosphobot cloud. This is the quickest way to train a model.
125125

126126
1. Enter the name of your dataset on Hugging Face (example: `PLB/simple-lego-pickup-mono-2`) in the **AI Training and Control** section.
127-
2. Click on **Train AI Model**. Your model starts training. Training can take up to 3 hours. Follow the training using the button **View trained models**. Your model is uploaded to HuggingFace [on the phospho-app account](https://huggingface.co/phospho-app).
128-
3. To control your robot with the trained model, go to the **Control your robot** section and enter the name of your model.
127+
2. Select the parameters you want to change or leave the default ones.
128+
3. Click on **Train AI Model**. Your model starts training. Training can take up to 3 hours. Follow the training using the button **View trained models**. Your model is uploaded to HuggingFace [on the phospho-app account](https://huggingface.co/phospho-app).
129+
4. To control your robot with the trained model, go to the **Control your robot** section and enter the name of your model.
129130

130131
![phosphobot training cloud](/assets/phosphobot-aitraining.png)
131132

@@ -158,7 +159,6 @@ title="Control your robot with GR00T-N1-2B"
158159
</Card>
159160
</CardGroup>
160161

161-
162162
### Train an ACT model locally with LeRobot
163163

164164
<Warning>

0 commit comments

Comments
 (0)