Dimios45 commited on
Commit
d28849a
·
verified ·
1 Parent(s): eb2249c

README: add me-full_25000_120 (final epoch-120 model, eval pending)

Browse files
Files changed (1) hide show
  1. README.md +8 -2
README.md CHANGED
@@ -26,6 +26,7 @@ diversity (NJD), computed via `kin_flow.cli.bench` on 10 held-out test scenes wi
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
 
30
  Single-embodiment models: `num_scenes=5000`. Multi-embodiment model: the paper's full
31
  configuration — `num_scenes=25000`, fp32, batch 5 scenes × 128 grasps, warmup-cosine LR
@@ -66,8 +67,13 @@ step; epoch numbers here are the corrected true dataset-pass count.
66
  | Shadow Hand | 22 | 79.2% | 0.243 |
67
  | **mean** | | **86.4%** | **0.212** |
68
 
69
- Approaching convergence (epoch 90 of ~120 needed, per the original repo's own README
70
- guidance) — newer-epoch checkpoints will be added as training progresses.
 
 
 
 
 
71
 
72
  ## Loading
73
 
 
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) | eval pending | eval pending | 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
 
67
  | Shadow Hand | 22 | 79.2% | 0.243 |
68
  | **mean** | | **86.4%** | **0.212** |
69
 
70
+ ### `me-full_25000_120` final model (epoch 120, training complete)
71
+
72
+ Training finished 2026-07-15 at exactly **120.0 true dataset passes** (gradient step
73
+ 302,000 = 3,000,000 scene-visits / 25,000 scenes), the convergence point recommended by
74
+ the original repo's README. Saved from the step-302,000 train-state checkpoint, so it
75
+ also contains optimizer state for further fine-tuning. Per-gripper eval is pending and
76
+ will be added here once complete.
77
 
78
  ## Loading
79