Skip to content

Commit c8a91ea

Browse files
authored
Update better simpler bridge result (#383)
* Update better simpler bridge result https://huggingface.co/youliangtan/gr00t-n1.5-bridge-posttrain/tree/main | Task | Before | After | Δ Change | | ----------------------------------- | ------------- | ------------- | ---------- | | widowx\_spoon\_on\_towel | 226/300 (75%) | 238/300 (79%) | 🔼 +4% | | widowx\_carrot\_on\_plate | 163/300 (54%) | 207/300 (69%) | 🔼 +15% | | widowx\_put\_eggplant\_in\_basket | 184/300 (61%) | 145/300 (48%) | 🔻 −13% | | widowx\_stack\_cube | 171/300 (57%) | 143/300 (48%) | 🔻 −9% | | widowx\_put\_eggplant\_in\_sink\*\* | 19/300 (6%) | 41/300 (14%) | 🔼 +8% | | widowx\_close\_drawer\*\* | 68/300 (23%) | 166/300 (55%) | 🔼 +32% | | widowx\_open\_drawer\*\* | 129/300 (43%) | 213/300 (71%) | 🔼 +28% | | **Average** | **46%** | **55%** | 🔼 **+9%** | * Update result for batch 90 * nit
1 parent 2196232 commit c8a91ea

1 file changed

Lines changed: 10 additions & 10 deletions

File tree

examples/SimplerEnv/README.md

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -15,14 +15,14 @@ Provided checkpoint: youliangtan/gr00t-n1.5-bridge-posttrain
1515

1616
| Task | Success rate (300) |
1717
| --------------------------------- | ------------------ |
18-
| widowx\_spoon\_on\_towel | 226/300 (75%) |
19-
| widowx\_carrot\_on\_plate | 163/300 (54%) |
20-
| widowx\_put\_eggplant\_in\_basket | 184/300 (61%) |
21-
| widowx\_stack\_cube | 171/300 (57%) |
22-
| widowx\_put\_eggplant\_in\_sink** | 19/300 (6%) |
23-
| widowx\_close\_drawer** | 68/300 (23%) |
24-
| widowx\_open\_drawer** | 129/300 (43%) |
25-
| **Average** | **46%** |
18+
| widowx\_spoon\_on\_towel | 246/300 (82%) |
19+
| widowx\_carrot\_on\_plate | 216/300 (72%) |
20+
| widowx\_put\_eggplant\_in\_basket | 189/300 (63%) |
21+
| widowx\_stack\_cube | 162/300 (54%) |
22+
| widowx\_put\_eggplant\_in\_sink** | 62/300 (21%) |
23+
| widowx\_close\_drawer** | 196/300 (65%) |
24+
| widowx\_open\_drawer** | 251/300 (84%) |
25+
| **Average** | **55%** |
2626

2727
**Denotes as "non-original" new simpler task [here](https://github.com/youliangtan/SimplerEnv)
2828

@@ -149,7 +149,7 @@ python scripts/gr00t_finetune.py \
149149
--dataset-path /tmp/bridge_orig_lerobot/ \
150150
--data_config examples.SimplerEnv.custom_data_config:BridgeDataConfig \
151151
--num-gpus 8 \
152-
--batch-size 64 \
152+
--batch-size 90 \
153153
--output-dir /tmp/bridge-checkpoints \
154154
--max-steps 60000
155155
```
@@ -164,4 +164,4 @@ python scripts/gr00t_finetune.py \
164164
--batch-size 128 \
165165
--output-dir /tmp/fractal-checkpoints/ \
166166
--max-steps 60000
167-
```
167+
```

0 commit comments

Comments
 (0)