Dimios45 commited on
Commit
75fd112
·
verified ·
1 Parent(s): 52a6096

README: show ME epochs as /120 (training target) instead of /500 config cap

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -23,10 +23,10 @@ diversity (NJD), computed via `kin_flow.cli.bench` on 10 held-out test scenes wi
23
  |---|---|---|---|---|---|---|
24
  | `se-panda_5000_170` | single-embodiment | Panda (2 DOF) | 170 / 500 | 97.8% | 0.293 | RTX 6000 Ada |
25
  | `se-shadow_5000_40` | single-embodiment | Shadow Hand (22 DOF) | 40 / 500 | 75.9% | 0.232 | RTX 6000 Ada |
26
- | `me-full_25000_5` | multi-embodiment | all 5 + z0 | 5 / 500 | 82.1% (mean) | 0.212 (mean) | MI300X (ROCm) |
27
- | `me-full_25000_30` | multi-embodiment | all 5 + z0 | 30 / 500 | 84.3% (mean) | 0.203 (mean) | MI300X (ROCm) |
28
- | `me-full_25000_90` | multi-embodiment | all 5 + z0 | 90 / 500 | 86.4% (mean) | 0.212 (mean) | MI300X (ROCm) |
29
- | `me-full_25000_120` | multi-embodiment | all 5 + z0 | **120** (target) | 86.1% (mean) | 0.209 (mean) | MI300X (ROCm) |
30
 
31
  Single-embodiment models: `num_scenes=5000`. Multi-embodiment model: the paper's full
32
  configuration — `num_scenes=25000`, fp32, batch 5 scenes × 128 grasps, warmup-cosine LR
 
23
  |---|---|---|---|---|---|---|
24
  | `se-panda_5000_170` | single-embodiment | Panda (2 DOF) | 170 / 500 | 97.8% | 0.293 | RTX 6000 Ada |
25
  | `se-shadow_5000_40` | single-embodiment | Shadow Hand (22 DOF) | 40 / 500 | 75.9% | 0.232 | RTX 6000 Ada |
26
+ | `me-full_25000_5` | multi-embodiment | all 5 + z0 | 5 / 120 | 82.1% (mean) | 0.212 (mean) | MI300X (ROCm) |
27
+ | `me-full_25000_30` | multi-embodiment | all 5 + z0 | 30 / 120 | 84.3% (mean) | 0.203 (mean) | MI300X (ROCm) |
28
+ | `me-full_25000_90` | multi-embodiment | all 5 + z0 | 90 / 120 | 86.4% (mean) | 0.212 (mean) | MI300X (ROCm) |
29
+ | `me-full_25000_120` | multi-embodiment | all 5 + z0 | **120 / 120** | 86.1% (mean) | 0.209 (mean) | MI300X (ROCm) |
30
 
31
  Single-embodiment models: `num_scenes=5000`. Multi-embodiment model: the paper's full
32
  configuration — `num_scenes=25000`, fp32, batch 5 scenes × 128 grasps, warmup-cosine LR