Update model card with step56000 recommendation
Browse files
README.md
CHANGED
|
@@ -12,11 +12,9 @@ language:
|
|
| 12 |
library_name: openpi
|
| 13 |
---
|
| 14 |
|
| 15 |
-
# Sortpackage
|
| 16 |
|
| 17 |
-
|
| 18 |
-
`round2_step85000/`。它来自 `sort-packages-combined-pi05-5epoch` 训练任务的
|
| 19 |
-
`round2_step85000` checkpoint,并已去除训练 optimizer 等训练参数,仅保留推理所需文件。
|
| 20 |
|
| 21 |
## 目标机器人
|
| 22 |
|
|
@@ -27,21 +25,45 @@ library_name: openpi
|
|
| 27 |
|
| 28 |
## 目标任务
|
| 29 |
|
| 30 |
-
|
| 31 |
|
| 32 |
-
##
|
| 33 |
|
| 34 |
```text
|
| 35 |
-
round2_step85000/
|
| 36 |
-
|
| 37 |
-
round2_step85000/assets/pi05_popcorn_21d/norm_stats.json
|
| 38 |
```
|
| 39 |
|
| 40 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
|
| 42 |
## Action 维度对应
|
| 43 |
|
| 44 |
-
策略输出为 21 维动作。在当前 G2_90d 新夹爪环境中使用直接布局,不需要旧
|
| 45 |
|
| 46 |
| 维度 | 关节/含义 |
|
| 47 |
|---:|---|
|
|
@@ -73,36 +95,22 @@ round2_step85000/assets/pi05_popcorn_21d/norm_stats.json
|
|
| 73 |
- `14:16`:90D 左右夹爪 2 维。
|
| 74 |
- `16:21`:G2 腰部/躯干 5 维,运行时对应 `idx01_body_joint1` 到 `idx05_body_joint5`。
|
| 75 |
|
| 76 |
-
## 推理
|
| 77 |
|
| 78 |
-
|
| 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 |
-
-
|
| 107 |
- 当前布局下不需要旧 motion mapper。
|
| 108 |
-
-
|
|
|
|
| 12 |
library_name: openpi
|
| 13 |
---
|
| 14 |
|
| 15 |
+
# Sortpackage G2_90d Policy Checkpoints
|
| 16 |
|
| 17 |
+
本仓库保存用于具身智能分拣包裹实验的 inference-only policy checkpoint。当前推荐候选是 `step56000/`。
|
|
|
|
|
|
|
| 18 |
|
| 19 |
## 目标机器人
|
| 20 |
|
|
|
|
| 25 |
|
| 26 |
## 目标任务
|
| 27 |
|
| 28 |
+
策略用于 `sorting_packages_continuous` / `Sort Packages` 任务:机器人在桌面或仓储工作台场景中抓取包裹/盒子,并尝试完成向右转腰和放置/分拣流程。
|
| 29 |
|
| 30 |
+
## Checkpoints
|
| 31 |
|
| 32 |
```text
|
| 33 |
+
round2_step85000/
|
| 34 |
+
step56000/
|
|
|
|
| 35 |
```
|
| 36 |
|
| 37 |
+
每个 checkpoint 均只包含推理所需文件,不包含 optimizer state 或训练分片缓存。
|
| 38 |
+
|
| 39 |
+
`step56000/` 文件结构:
|
| 40 |
+
|
| 41 |
+
```text
|
| 42 |
+
step56000/model.safetensors
|
| 43 |
+
step56000/metadata.pt
|
| 44 |
+
step56000/assets/pi05_popcorn_21d/norm_stats.json
|
| 45 |
+
step56000/assets/pi05_sort_packages_combined_21d/norm_stats.json
|
| 46 |
+
```
|
| 47 |
+
|
| 48 |
+
其中 `pi05_sort_packages_combined_21d` 是原始 asset 名,`pi05_popcorn_21d` 是为了兼容当前 runtime 默认查找路径而保留的一份同内容 `norm_stats.json`。
|
| 49 |
+
|
| 50 |
+
## 当前模型选择结论
|
| 51 |
+
|
| 52 |
+
按同一规格评测:自然光 6500K / 1000,prompt 为 `Sort Packages`,新夹爪环境,无 mapper,seeds 为 `1 11 21 31 41 51 61 71`,每个权重 32 局。
|
| 53 |
+
|
| 54 |
+
| 权重 | 抓取 | E2E | upright | inside | 抓后右转腰部 |
|
| 55 |
+
|---|---:|---:|---:|---:|---:|
|
| 56 |
+
| `round2_step85000` | 4/32 | 0/32 | 1 | 0 | 0 |
|
| 57 |
+
| `step56000` | 3/32 | 0/32 | 2 | 1 | 1 |
|
| 58 |
+
| `step64000` | 2/32 | 0/32 | 1 | 0 | 0 |
|
| 59 |
+
|
| 60 |
+
明确选择:`step56000` 是当前最佳候选。它的抓取率低于旧 `round2_step85000`,但它是唯一出现 `inside=1` 和抓取后明显右转腰部的权重,更接近完整分拣流程。`round2_step85000` 适合作为抓取率较高的基线;`step64000` 弱于 `step56000`。
|
| 61 |
+
|
| 62 |
+
当前评测仍未出现 E2E 成功,因此这些权重更适合作为研究和继续调试的 checkpoint,而不是稳定生产策略。
|
| 63 |
|
| 64 |
## Action 维度对应
|
| 65 |
|
| 66 |
+
策略输出为 21 维动作。在当前 G2_90d 新夹爪环境中使用直接布局,不需要旧 motion mapper,也不需要对夹爪做 relabel。
|
| 67 |
|
| 68 |
| 维度 | 关节/含义 |
|
| 69 |
|---:|---|
|
|
|
|
| 95 |
- `14:16`:90D 左右夹爪 2 维。
|
| 96 |
- `16:21`:G2 腰部/躯干 5 维,运行时对应 `idx01_body_joint1` 到 `idx05_body_joint5`。
|
| 97 |
|
| 98 |
+
## 推理说明
|
| 99 |
|
| 100 |
+
容器内可将目标 checkpoint 挂载到 runtime 默认目录:
|
| 101 |
|
| 102 |
```text
|
| 103 |
/home/syf/popcorn_pi05_best_9000
|
| 104 |
```
|
| 105 |
|
| 106 |
+
策略服务 prompt 建议固定为:
|
| 107 |
|
| 108 |
```text
|
| 109 |
Sort Packages
|
| 110 |
```
|
| 111 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 112 |
## 注意事项
|
| 113 |
|
| 114 |
+
- 面向 `G2_90d` 新夹爪布局;不要按旧 `idx41/idx81` 夹爪接口使用。
|
| 115 |
- 当前布局下不需要旧 motion mapper。
|
| 116 |
+
- `step56000` 是当前最佳整体候选,但 E2E 仍为 0/32,需要继续前台复现和定位。
|