Update model card for Sortpackage round2 checkpoint
Browse files
README.md
CHANGED
|
@@ -1,3 +1,108 @@
|
|
| 1 |
---
|
| 2 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
+
tags:
|
| 3 |
+
- robotics
|
| 4 |
+
- embodied-ai
|
| 5 |
+
- policy
|
| 6 |
+
- geniesim
|
| 7 |
+
- g2-90d
|
| 8 |
+
- package-sorting
|
| 9 |
+
language:
|
| 10 |
+
- zh
|
| 11 |
+
- en
|
| 12 |
+
library_name: openpi
|
| 13 |
---
|
| 14 |
+
|
| 15 |
+
# Sortpackage Round2 Step85000 Policy
|
| 16 |
+
|
| 17 |
+
这是一个用于具身智能分拣包裹实验的推理权重,上传目录为
|
| 18 |
+
`round2_step85000/`。它来自 `sort-packages-combined-pi05-5epoch` 训练任务的
|
| 19 |
+
`round2_step85000` checkpoint,并已去除训练 optimizer 等训练参数,仅保留推理所需文件。
|
| 20 |
+
|
| 21 |
+
## 目标机器人
|
| 22 |
+
|
| 23 |
+
- 机器人配置:`G2_90d`
|
| 24 |
+
- 夹爪:90D 新夹爪
|
| 25 |
+
- 运行环境:GenieSim / G2 90D gripper 仿真环境
|
| 26 |
+
- 推荐策略提示词:`Sort Packages`
|
| 27 |
+
|
| 28 |
+
## 目标任务
|
| 29 |
+
|
| 30 |
+
该策略用于 `sorting_packages_continuous` / `Sort Packages` 任务:机器人需要在桌面或仓储工作台场景中抓取包裹/盒子,并完成包裹分拣流程。当前 checkpoint 是为了评估 G2_90d 机器人在分拣包裹任务上的迁移与推理效果而保存的 inference-only 权重。
|
| 31 |
+
|
| 32 |
+
## 文件
|
| 33 |
+
|
| 34 |
+
```text
|
| 35 |
+
round2_step85000/model.safetensors
|
| 36 |
+
round2_step85000/metadata.pt
|
| 37 |
+
round2_step85000/assets/pi05_popcorn_21d/norm_stats.json
|
| 38 |
+
```
|
| 39 |
+
|
| 40 |
+
其中 `model.safetensors` 是策略权重,`metadata.pt` 与 `norm_stats.json` 为推理所需元数据和归一化统计。仓库中没有上传 optimizer state 或完整训练中间状态。
|
| 41 |
+
|
| 42 |
+
## Action 维度对应
|
| 43 |
+
|
| 44 |
+
策略输出为 21 维动作。在当前 G2_90d 新夹爪环境中使用直接布局,不需要旧的 motion mapper,也不需要对夹爪做 relabel。
|
| 45 |
+
|
| 46 |
+
| 维度 | 关节/含义 |
|
| 47 |
+
|---:|---|
|
| 48 |
+
| 0 | `idx21_arm_l_joint1` |
|
| 49 |
+
| 1 | `idx22_arm_l_joint2` |
|
| 50 |
+
| 2 | `idx23_arm_l_joint3` |
|
| 51 |
+
| 3 | `idx24_arm_l_joint4` |
|
| 52 |
+
| 4 | `idx25_arm_l_joint5` |
|
| 53 |
+
| 5 | `idx26_arm_l_joint6` |
|
| 54 |
+
| 6 | `idx27_arm_l_joint7` |
|
| 55 |
+
| 7 | `idx61_arm_r_joint1` |
|
| 56 |
+
| 8 | `idx62_arm_r_joint2` |
|
| 57 |
+
| 9 | `idx63_arm_r_joint3` |
|
| 58 |
+
| 10 | `idx64_arm_r_joint4` |
|
| 59 |
+
| 11 | `idx65_arm_r_joint5` |
|
| 60 |
+
| 12 | `idx66_arm_r_joint6` |
|
| 61 |
+
| 13 | `idx67_arm_r_joint7` |
|
| 62 |
+
| 14 | `idx32_gripper_l_outer_joint1` |
|
| 63 |
+
| 15 | `idx72_gripper_r_outer_joint1` |
|
| 64 |
+
| 16 | `idx01_body_joint1` |
|
| 65 |
+
| 17 | `idx02_body_joint2` |
|
| 66 |
+
| 18 | `idx03_body_joint3` |
|
| 67 |
+
| 19 | `idx04_body_joint4` |
|
| 68 |
+
| 20 | `idx05_body_joint5` |
|
| 69 |
+
|
| 70 |
+
维度分组:
|
| 71 |
+
|
| 72 |
+
- `0:14`:G2 左右双臂 14 个关节,左臂 7 维后接右臂 7 维。
|
| 73 |
+
- `14:16`:90D 左右夹爪 2 维。
|
| 74 |
+
- `16:21`:G2 腰部/躯干 5 维,运行时对应 `idx01_body_joint1` 到 `idx05_body_joint5`。
|
| 75 |
+
|
| 76 |
+
## 推理使用说明
|
| 77 |
+
|
| 78 |
+
该权重按 policy server 默认 checkpoint 目录组织。容器内可挂载到:
|
| 79 |
+
|
| 80 |
+
```text
|
| 81 |
+
/home/syf/popcorn_pi05_best_9000
|
| 82 |
+
```
|
| 83 |
+
|
| 84 |
+
启动策略服务时建议使用提示词:
|
| 85 |
+
|
| 86 |
+
```text
|
| 87 |
+
Sort Packages
|
| 88 |
+
```
|
| 89 |
+
|
| 90 |
+
## 已知验证状态
|
| 91 |
+
|
| 92 |
+
在 Syf 隔离的 GenieSim G2_90d 自然光环境中做过多 seed 仿真对比。合并 8 个 seed、每个权重 32 局的结果中,`round2_step85000` 在候选权重里抓取率最高:
|
| 93 |
+
|
| 94 |
+
```text
|
| 95 |
+
pickup: 4/32
|
| 96 |
+
e2e: 0/32
|
| 97 |
+
upright: 1
|
| 98 |
+
inside: 0
|
| 99 |
+
right_turn_after_pickup: 0
|
| 100 |
+
```
|
| 101 |
+
|
| 102 |
+
因此该权重目前更适合作为分拣包裹任务的研究/对比 checkpoint,而不是已验证可直接部署的稳定生产策略。
|
| 103 |
+
|
| 104 |
+
## 注意事项
|
| 105 |
+
|
| 106 |
+
- 该 checkpoint 面向 `G2_90d` 新夹爪布局;不要按旧 `idx41/idx81` 夹爪接口使用。
|
| 107 |
+
- 当前布局下不需要旧 motion mapper。
|
| 108 |
+
- 任务提示词应与训练/评测保持一致,使用 `Sort Packages`。
|