RB3159 commited on
Commit
cd6599e
·
verified ·
1 Parent(s): 3de7d3a

Rewrite model card: full AWBC/LR-soup lineage, dataset & sibling links (team INHA-UNITED)

Browse files
Files changed (1) hide show
  1. README.md +32 -5
README.md CHANGED
@@ -1,19 +1,46 @@
1
  ---
2
  license: apache-2.0
 
 
3
  tags:
4
  - robotics
5
  - vla
6
  - pi0.5
 
7
  - rby1
 
 
 
8
  - robocup
9
  ---
10
 
11
  # Doing Laundry — RoboCup 2026 Incheon
12
 
13
- **Team: INHA-UNITED**
14
 
15
- RBY1 로봇의 빨래 개기(fold the t-shirt) 태스크용 Pi0.5 VLA 체크포인트 (Model Arithmetic greedy soup 병합본, `M0_lr_soup`).
16
 
17
- - Base: `pi05_rby1_fold_v11_1prompt_f2r03_awbc`
18
- - Merge: LR sweep (lr1×2 + lr2 + lr4) greedy weight soup
19
- - Prompt: `"fold the t-shirt, Advantage: positive"`
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
+ library_name: openpi
4
+ pipeline_tag: robotics
5
  tags:
6
  - robotics
7
  - vla
8
  - pi0.5
9
+ - openpi
10
  - rby1
11
+ - fold
12
+ - awbc
13
+ - model-soup
14
  - robocup
15
  ---
16
 
17
  # Doing Laundry — RoboCup 2026 Incheon
18
 
19
+ > **Team INHA-UNITED** · RB-Y1 T-shirt folding policy for the RoboCup 2026 @Home league (Incheon).
20
 
21
+ A π0.5 vision-language-action policy for Rainbow Robotics RB-Y1 T-shirt folding, deployed as the competition model. This is a **greedy model soup** (`M0_lr_soup`) — a weight average of a learning-rate sweep over the AWBC fold family, picked to maximize robustness on the real robot.
22
 
23
+ - **Method:** Model Arithmetic **greedy weight soup** of a 3-way LR sweep, all warm-started from the [v11 AWBC](https://huggingface.co/RB3159/pi05-rby1-fold-awbc) checkpoint (step 69999) and continued to step 99999. Members and soup weights: `lr1` (2.5e-6) ×2, `lr2` (5e-6) ×1, `lr4` (1e-5) ×1 — `lr1` was selected twice by greedy souping.
24
+ - **Training objective:** Advantage-Weighted Behavior Cloning (AWBC) following χ₀ (kai0) — per-frame `awbc_weight` from the [advantage estimator](https://huggingface.co/RB3159/rby1-fold-advantage-estimator)'s relative advantage `V(s+H) − V(s)`, thresholded at a fixed global p70 cutoff (~30% positive). See [arXiv:2602.09021](https://arxiv.org/abs/2602.09021), [OpenDriveLab/kai0](https://github.com/OpenDriveLab/kai0).
25
+ - **Prompt:** single task, `fold the t-shirt, Advantage: positive` (hold at inference).
26
+ - **Architecture:** π0.5 (`pi05`, PaliGemma `gemma_2b` + action expert `gemma_300m`, action horizon 50).
27
+ - **Lineage:** [pi05-rby1-fold-base](https://huggingface.co/RB3159/pi05-rby1-fold-base) (mir280) → [3stage](https://huggingface.co/RB3159/pi05-rby1-fold-3stage) → [v11 AWBC](https://huggingface.co/RB3159/pi05-rby1-fold-awbc) → LR sweep → **this soup**.
28
+ - **Training data:** 432 episodes — [rby1-fold-v1.1](https://huggingface.co/datasets/RB3159/rby1-fold-v1.1) master + left/right mirror + weak time-scale augmentation + DAgger interventions, with a per-frame `awbc_weight` column, img224.
29
+ - **Action space:** 16-dim — `[R_arm(7), R_gripper(1), L_arm(7), L_gripper(1)]`, delta on the 7-DoF arms, absolute on the grippers.
30
+ - **Robot:** Rainbow Robotics RB-Y1 (dual-arm).
31
+
32
+ ## Why a soup
33
+
34
+ The three LR-sweep runs share an init and a data distribution but converge to different basins. Averaging their weights (greedy soup) keeps the shared skill while smoothing over run-specific overfitting — no extra inference cost, and empirically the safest single model to deploy under competition conditions.
35
+
36
+ ## Files
37
+
38
+ `params/` (orbax inference weights) + `assets/.../norm_stats.json` + `_CHECKPOINT_METADATA`. Trained/merged with [openpi](https://github.com/Physical-Intelligence/openpi).
39
+
40
+ ## Usage
41
+
42
+ ```bash
43
+ python scripts/serve_policy.py --config pi05_rby1_fold_v11_1prompt_f2r03_awbc --checkpoint RB3159/Doing-Laundry-Robocup2026-Incheon
44
+ ```
45
+
46
+ Hold the prompt to `fold the t-shirt, Advantage: positive`.