Update README.md
Browse files
README.md
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
#
|
| 2 |
|
| 3 |
This is a PyTorch version of the PI0 pi0_base model, converted from the original JAX/Flax implementation.
|
| 4 |
|
|
@@ -8,9 +8,6 @@ This is a PyTorch version of the PI0 pi0_base model, converted from the original
|
|
| 8 |
- **Model Type**: PI0
|
| 9 |
- **Domain**: Base model (general purpose)
|
| 10 |
- **Precision**: 32-bit floating point (fp32)
|
| 11 |
-
- **Action Dimension**: 32
|
| 12 |
-
- **Action Horizon**: 50
|
| 13 |
-
- **Max Token Length**: 48
|
| 14 |
- **Vision Model**: PaliGemma (gemma_2b)
|
| 15 |
- **Action Expert**: gemma_300m
|
| 16 |
|
|
@@ -61,7 +58,6 @@ The model consists of:
|
|
| 61 |
|
| 62 |
This model was trained on robotics datasets appropriate for its domain:
|
| 63 |
- **DROID models**: Trained on diverse robot manipulation data
|
| 64 |
-
- **ALOHA models**: Trained on bimanual manipulation tasks
|
| 65 |
- **LIBERO models**: Trained on diverse tabletop manipulation scenarios
|
| 66 |
- **Base models**: Trained on general robotics datasets
|
| 67 |
|
|
|
|
| 1 |
+
# π₀ - Base
|
| 2 |
|
| 3 |
This is a PyTorch version of the PI0 pi0_base model, converted from the original JAX/Flax implementation.
|
| 4 |
|
|
|
|
| 8 |
- **Model Type**: PI0
|
| 9 |
- **Domain**: Base model (general purpose)
|
| 10 |
- **Precision**: 32-bit floating point (fp32)
|
|
|
|
|
|
|
|
|
|
| 11 |
- **Vision Model**: PaliGemma (gemma_2b)
|
| 12 |
- **Action Expert**: gemma_300m
|
| 13 |
|
|
|
|
| 58 |
|
| 59 |
This model was trained on robotics datasets appropriate for its domain:
|
| 60 |
- **DROID models**: Trained on diverse robot manipulation data
|
|
|
|
| 61 |
- **LIBERO models**: Trained on diverse tabletop manipulation scenarios
|
| 62 |
- **Base models**: Trained on general robotics datasets
|
| 63 |
|