AdrianLlopart commited on
Commit
c5ccfcf
·
0 Parent(s):

Duplicate from AdrianLlopart/rskill-act-aloha-insertion

Browse files
Files changed (4) hide show
  1. .gitattributes +35 -0
  2. README.md +67 -0
  3. eval/aloha_insertion.json +89 -0
  4. rskill.yaml +71 -0
.gitattributes ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - OpenRAL
4
+ - rskill
5
+ - act
6
+ - lerobot
7
+ - aloha
8
+ - bimanual
9
+ - manipulation
10
+ - insertion
11
+ license: mit
12
+ language:
13
+ - en
14
+ ---
15
+
16
+ # rskill-act-aloha-insertion
17
+
18
+ > **OpenRAL rSkill (custom example)** — ACT (Action Chunking Transformer)
19
+ > finetuned on the ALOHA bimanual **peg-insertion** task, packaged for
20
+ > `OpenRAL`.
21
+
22
+ This package wraps
23
+ [`lerobot/act_aloha_sim_insertion_human`](https://huggingface.co/lerobot/act_aloha_sim_insertion_human)
24
+ with a `rskill.yaml` manifest that adds capability checking, license
25
+ surfacing, latency budgets, and local registry integration. It does
26
+ **not** copy model weights.
27
+
28
+ It is the harder sibling of [`rskill-act-aloha`](../act-aloha) (cube
29
+ transfer) and demonstrates how a single packaging format covers multiple
30
+ task-specific checkpoints from the same paper. The runnable demo lives at
31
+ `examples/sim/custom_act_aloha_insertion.yaml` and is wired into the
32
+ top-level `just sim-custom` recipe.
33
+
34
+ ## Upstream model
35
+
36
+ | Field | Value |
37
+ | --- | --- |
38
+ | Source repo | [`lerobot/act_aloha_sim_insertion_human`](https://huggingface.co/lerobot/act_aloha_sim_insertion_human) |
39
+ | Architecture | Action Chunking Transformer (~52M params, chunk=100) |
40
+ | Task | gym-aloha `AlohaInsertion-v0` (bimanual peg-in-socket) |
41
+ | License | MIT |
42
+ | Paper | Zhao et al., 2023 — *Learning Fine-Grained Bimanual Manipulation with Low-Cost Hardware* ([arXiv 2304.13705](https://arxiv.org/abs/2304.13705)) |
43
+
44
+ ## Why no `eval/` block?
45
+
46
+ This skill is shipped as a **custom-example** package, not as a
47
+ reproduced benchmark entry. The paper's headline number for sim ALOHA
48
+ insertion is markedly lower than the cube-transfer figure (the task is
49
+ harder and the upstream protocol uses different camera intrinsics). We
50
+ deliberately omit `eval/` rather than copy paper numbers without an
51
+ internal reproduction; per CLAUDE.md §6.4 that omission must be
52
+ documented — this section is that documentation. Add `eval/aloha_insertion.json`
53
+ once a local reproduction lands.
54
+
55
+ ## Run it
56
+
57
+ ```bash
58
+ just sim-custom
59
+ ```
60
+
61
+ …which is equivalent to:
62
+
63
+ ```bash
64
+ MUJOCO_GL=egl uv run --group sim ral sim run \
65
+ --config examples/sim/custom_act_aloha_insertion.yaml \
66
+ --save-video example_videos
67
+ ```
eval/aloha_insertion.json ADDED
@@ -0,0 +1,89 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": "1",
3
+ "source": {
4
+ "paper": "https://arxiv.org/abs/2304.13705",
5
+ "arxiv": "https://arxiv.org/abs/2304.13705",
6
+ "model_variant": "act",
7
+ "evaluated_by": "OpenRAL:ral benchmark run",
8
+ "reproduced_locally": true,
9
+ "reproduction_planned": null,
10
+ "reproduction_cli": "ral benchmark run --suite aloha_insertion --rskill rskill://act-aloha-insertion",
11
+ "table": null,
12
+ "status": "reproduced"
13
+ },
14
+ "benchmark": {
15
+ "name": "ALOHA bimanual peg insertion (gym-aloha)",
16
+ "dataset": null,
17
+ "protocol": "50 episodes per task, success_key=is_success, max_steps=400",
18
+ "robot": "aloha_bimanual",
19
+ "simulator": "gym-aloha (MuJoCo)"
20
+ },
21
+ "eval_config": {
22
+ "n_episodes": 50,
23
+ "seeds": [
24
+ 0,
25
+ 1,
26
+ 2,
27
+ 3,
28
+ 4,
29
+ 5,
30
+ 6,
31
+ 7,
32
+ 8,
33
+ 9,
34
+ 10,
35
+ 11,
36
+ 12,
37
+ 13,
38
+ 14,
39
+ 15,
40
+ 16,
41
+ 17,
42
+ 18,
43
+ 19,
44
+ 20,
45
+ 21,
46
+ 22,
47
+ 23,
48
+ 24,
49
+ 25,
50
+ 26,
51
+ 27,
52
+ 28,
53
+ 29,
54
+ 30,
55
+ 31,
56
+ 32,
57
+ 33,
58
+ 34,
59
+ 35,
60
+ 36,
61
+ 37,
62
+ 38,
63
+ 39,
64
+ 40,
65
+ 41,
66
+ 42,
67
+ 43,
68
+ 44,
69
+ 45,
70
+ 46,
71
+ 47,
72
+ 48,
73
+ 49
74
+ ],
75
+ "success_key": "is_success",
76
+ "max_steps": 400,
77
+ "vla_id": "act",
78
+ "weights_uri": "rskill://rskills/act-aloha-insertion"
79
+ },
80
+ "results": {
81
+ "aloha_insertion/0_success_rate": 0.2,
82
+ "avg_success_rate": 0.2,
83
+ "n_tasks": 1,
84
+ "n_episodes_per_task": 50,
85
+ "n_episodes_total": 50,
86
+ "mean_step_latency_ms_avg": 10.16975808690558
87
+ },
88
+ "baselines": {}
89
+ }
rskill.yaml ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # rSkill manifest — OpenRAL packaging format V1 (CLAUDE.md §6.4)
2
+ # Wraps: lerobot/act_aloha_sim_insertion_human (MIT)
3
+ # Paper: Zhao et al., 2023 — Action Chunking Transformer.
4
+ #
5
+ # Sibling of skills/act-aloha (which wraps the transfer-cube checkpoint).
6
+ # This package targets the harder bimanual peg-in-socket *insertion* task
7
+ # from the same paper. Both skills share the ALOHA 14-DoF action space and
8
+ # the single 480x640 top camera contract.
9
+
10
+ schema_version: "1"
11
+
12
+ name: "AdrianLlopart/rskill-act-aloha-insertion"
13
+ version: "0.1.0"
14
+ license: "mit"
15
+ role: "s1"
16
+
17
+ model_family: "act"
18
+
19
+ embodiment_tags:
20
+ - "aloha"
21
+
22
+ capabilities_required: {}
23
+
24
+ sensors_required:
25
+ - modality: "rgb"
26
+ vla_feature_key: "observation.images.top"
27
+ min_width: 640
28
+ min_height: 480
29
+
30
+ # Output side (ADR-0013). For the canonical aloha bimanual embodiment the
31
+ # loader auto-fills n_dof (14) + vla_action_key from
32
+ # robots/aloha_bimanual/robot.yaml.
33
+ actuators_required:
34
+ - kind: "joint_position"
35
+
36
+ runtime: "pytorch"
37
+
38
+ quantization:
39
+ dtype: "fp32"
40
+ backend: "pytorch"
41
+
42
+ weights_uri: "hf://lerobot/act_aloha_sim_insertion_human"
43
+
44
+ chunk_size: 100
45
+
46
+ latency_budget:
47
+ # Same ACT architecture as the cube-transfer sibling; reference-host
48
+ # measurements match within noise. The 25 ms ceiling keeps the
49
+ # canonical tolerance_pct=100 → 50 ms test ceiling.
50
+ per_chunk_ms: 25.0
51
+ warmup_ms: 5000.0
52
+ load_ms: 10000.0
53
+
54
+ fallback_skill_id: null
55
+
56
+ # Headline success rate from skills/act-aloha-insertion/eval/aloha_insertion.json.
57
+ benchmarks:
58
+ aloha_insertion: 0.20
59
+
60
+ policy_id: "act"
61
+
62
+ paper_url: "https://arxiv.org/abs/2304.13705"
63
+ source_repo: "hf://lerobot/act_aloha_sim_insertion_human"
64
+
65
+ description: >
66
+ ACT (~52M params, chunk=100) finetuned on the ALOHA bimanual
67
+ sim-insertion demonstration set. Like the transfer-cube sibling, the
68
+ published checkpoint predates lerobot's PolicyProcessorPipeline
69
+ migration and ships without normalisation buffers. Insertion is the
70
+ harder ALOHA task in the original paper; the 0.20 success rate
71
+ reflects that.