lakomchik commited on
Commit
76af586
·
verified ·
1 Parent(s): e5aca07

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -18
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