Update README.md
Browse files
README.md
CHANGED
|
@@ -4,7 +4,7 @@ license: apache-2.0
|
|
| 4 |
language:
|
| 5 |
- en
|
| 6 |
base_model:
|
| 7 |
-
- SberRoboticsCenter/GreenVLA-5b-base
|
| 8 |
pipeline_tag: robotics
|
| 9 |
tags:
|
| 10 |
- robotics
|
|
@@ -19,7 +19,7 @@ tags:
|
|
| 19 |
datasets:
|
| 20 |
- IPEC-COMMUNITY/bridge_orig_lerobot
|
| 21 |
model-index:
|
| 22 |
-
- name: GreenVLA-5b-R1-bridge
|
| 23 |
results:
|
| 24 |
- task:
|
| 25 |
type: robotics
|
|
@@ -38,7 +38,7 @@ model-index:
|
|
| 38 |
|
| 39 |
<div align="center">
|
| 40 |
|
| 41 |
-
# GreenVLA-5b-R1-bridge
|
| 42 |
|
| 43 |
### Embodiment-Adapted VLA for Bridge (WidowX)
|
| 44 |
|
|
@@ -54,13 +54,15 @@ model-index:
|
|
| 54 |
|
| 55 |
## Overview
|
| 56 |
|
| 57 |
-
**GreenVLA-5b-R1-bridge** is the R1 (embodiment-adapted) checkpoint of the [Green-VLA](https://arxiv.org/abs/2602.00919) family, fine-tuned on the [Bridge](https://huggingface.co/datasets/IPEC-COMMUNITY/bridge_orig_lerobot) dataset for the WidowX robot arm.
|
| 58 |
|
| 59 |
-
Starting from the [GreenVLA-5b-base](https://huggingface.co/SberRoboticsCenter/GreenVLA-5b-base) pretrained checkpoint, this model was adapted via supervised fine-tuning (R1 stage) to the Bridge embodiment, achieving strong manipulation performance on the SimplerEnv benchmark.
|
| 60 |
|
| 61 |
## Evaluation
|
| 62 |
|
| 63 |
-
Evaluated on **SimplerEnv WidowX (Bridge)** benchmark with default episode length
|
|
|
|
|
|
|
| 64 |
|
| 65 |
### Partial Success Rate
|
| 66 |
|
|
@@ -86,7 +88,7 @@ Evaluated on **SimplerEnv WidowX (Bridge)** benchmark with default episode lengt
|
|
| 86 |
|
| 87 |
| | Details |
|
| 88 |
|---|---|
|
| 89 |
-
| **Base checkpoint** | [GreenVLA-5b-base](https://huggingface.co/SberRoboticsCenter/GreenVLA-5b-base) |
|
| 90 |
| **Stage** | R1 — Embodiment-specific adaptation |
|
| 91 |
| **Method** | Supervised fine-tuning |
|
| 92 |
| **Dataset** | [IPEC-COMMUNITY/bridge_orig_lerobot](https://huggingface.co/datasets/IPEC-COMMUNITY/bridge_orig_lerobot) |
|
|
@@ -116,7 +118,7 @@ from lerobot.common.utils.torch_observation import (
|
|
| 116 |
|
| 117 |
# 1. Load policy and transforms.
|
| 118 |
policy, input_transforms, output_transforms = load_pretrained_policy(
|
| 119 |
-
"SberRoboticsCenter/GreenVLA-5b-R1-bridge",
|
| 120 |
data_config_name="bridge",
|
| 121 |
)
|
| 122 |
policy.to("cuda").eval()
|
|
@@ -145,16 +147,6 @@ actions = output_transforms(
|
|
| 145 |
|
| 146 |
See [`examples/example_inference_bridge.py`](https://github.com/greenvla/GreenVLA/blob/main/examples/example_inference_bridge.py) for the full runnable script with argument parsing.
|
| 147 |
|
| 148 |
-
## Model Family
|
| 149 |
-
|
| 150 |
-
| Model | Stage | Params | Description | Link |
|
| 151 |
-
|-------|:-----:|:------:|-------------|:----:|
|
| 152 |
-
| **GreenVLA-2b-base** | Base | 2B | Base pretrained (lightweight) | [Hub](https://huggingface.co/SberRoboticsCenter/GreenVLA-2b-base) |
|
| 153 |
-
| **GreenVLA-5b-base** | Base | 5B | Base pretrained (recommended) | [Hub](https://huggingface.co/SberRoboticsCenter/GreenVLA-5b-base) |
|
| 154 |
-
| **GreenVLA-5b-R1-bridge** | R1 | 5B | Fine-tuned on Bridge (WidowX) | You are here |
|
| 155 |
-
| **GreenVLA-5b-R2-bridge** | R2 | 5B | RL-aligned on Bridge (WidowX) | [Hub](https://huggingface.co/SberRoboticsCenter/GreenVLA-5b-R2-bridge) |
|
| 156 |
-
| **GreenVLA-5b-R1-fractal** | R1 | 5B | Fine-tuned on Fractal (Google Robot) | [Hub](https://huggingface.co/SberRoboticsCenter/GreenVLA-5b-R1-fractal) |
|
| 157 |
-
|
| 158 |
## Citation
|
| 159 |
|
| 160 |
```bibtex
|
|
|
|
| 4 |
language:
|
| 5 |
- en
|
| 6 |
base_model:
|
| 7 |
+
- SberRoboticsCenter/GreenVLA-5b-base-stride-1
|
| 8 |
pipeline_tag: robotics
|
| 9 |
tags:
|
| 10 |
- robotics
|
|
|
|
| 19 |
datasets:
|
| 20 |
- IPEC-COMMUNITY/bridge_orig_lerobot
|
| 21 |
model-index:
|
| 22 |
+
- name: GreenVLA-5b-stride-1-R1-bridge
|
| 23 |
results:
|
| 24 |
- task:
|
| 25 |
type: robotics
|
|
|
|
| 38 |
|
| 39 |
<div align="center">
|
| 40 |
|
| 41 |
+
# GreenVLA-5b-stride-1-R1-bridge
|
| 42 |
|
| 43 |
### Embodiment-Adapted VLA for Bridge (WidowX)
|
| 44 |
|
|
|
|
| 54 |
|
| 55 |
## Overview
|
| 56 |
|
| 57 |
+
**GreenVLA-5b-stride-1-R1-bridge** is the R1 (embodiment-adapted) checkpoint of the [Green-VLA](https://arxiv.org/abs/2602.00919) family, fine-tuned on the [Bridge](https://huggingface.co/datasets/IPEC-COMMUNITY/bridge_orig_lerobot) dataset for the WidowX robot arm.
|
| 58 |
|
| 59 |
+
Starting from the [GreenVLA-5b-base-stride-1](https://huggingface.co/SberRoboticsCenter/GreenVLA-5b-base-stride-1) pretrained checkpoint, this model was adapted via supervised fine-tuning (R1 stage) to the Bridge embodiment, achieving strong manipulation performance on the SimplerEnv benchmark.
|
| 60 |
|
| 61 |
## Evaluation
|
| 62 |
|
| 63 |
+
Evaluated on **SimplerEnv WidowX (Bridge)** benchmark with default episode length.
|
| 64 |
+
|
| 65 |
+
> **Note:** Bridge benchmark results can vary up to ±6% between runs. We recommend averaging over multiple evaluation runs for reliable comparisons.
|
| 66 |
|
| 67 |
### Partial Success Rate
|
| 68 |
|
|
|
|
| 88 |
|
| 89 |
| | Details |
|
| 90 |
|---|---|
|
| 91 |
+
| **Base checkpoint** | [GreenVLA-5b-base-stride-1](https://huggingface.co/SberRoboticsCenter/GreenVLA-5b-base-stride-1) |
|
| 92 |
| **Stage** | R1 — Embodiment-specific adaptation |
|
| 93 |
| **Method** | Supervised fine-tuning |
|
| 94 |
| **Dataset** | [IPEC-COMMUNITY/bridge_orig_lerobot](https://huggingface.co/datasets/IPEC-COMMUNITY/bridge_orig_lerobot) |
|
|
|
|
| 118 |
|
| 119 |
# 1. Load policy and transforms.
|
| 120 |
policy, input_transforms, output_transforms = load_pretrained_policy(
|
| 121 |
+
"SberRoboticsCenter/GreenVLA-5b-stride-1-R1-bridge",
|
| 122 |
data_config_name="bridge",
|
| 123 |
)
|
| 124 |
policy.to("cuda").eval()
|
|
|
|
| 147 |
|
| 148 |
See [`examples/example_inference_bridge.py`](https://github.com/greenvla/GreenVLA/blob/main/examples/example_inference_bridge.py) for the full runnable script with argument parsing.
|
| 149 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 150 |
## Citation
|
| 151 |
|
| 152 |
```bibtex
|