Drop the training-step numbers from the model card
Browse files
README.md
CHANGED
|
@@ -33,8 +33,7 @@ simulation environments live in the benchmark environment repository.
|
|
| 33 |
|
| 34 |
Every method's weights are called **`ckpt`** whichever trainer wrote them — a
|
| 35 |
directory for the JAX/orbax and LeRobot policies, a single file for Vid2Robot
|
| 36 |
-
(`ckpt.pt`) and UniSkill (`ckpt.pth`).
|
| 37 |
-
to is in the table below, not in the path.
|
| 38 |
|
| 39 |
`<condition>` is the training suite, and it is the **top** level: **`nd/`** = no
|
| 40 |
distraction (clean scenes), **`wd/`** = with distraction (distractors in both
|
|
@@ -47,23 +46,13 @@ bank.
|
|
| 47 |
|
| 48 |
## Checkpoints
|
| 49 |
|
| 50 |
-
| Paper label | Method dir | Weights |
|
| 51 |
-
|---|---|---|---|
|
| 52 |
-
| VLM + π0.5 | `vlm_pi05` | `ckpt/` |
|
| 53 |
-
| Vid. Cond. π0.5 (Uniform) | `vid_cond_pi05_uniform` | `ckpt/` |
|
| 54 |
-
| Vid. Cond. π0.5 (Keyframes) | `vid_cond_pi05_keyframes` | `ckpt/` |
|
| 55 |
-
| Vid2Robot | `vid2robot` | `ckpt.pt` |
|
| 56 |
-
| UniSkill | `uniskill_policy` | `ckpt.pth` |
|
| 57 |
-
|
| 58 |
-
**The WD keyframes policy is from step 20,000, not 50,000.** That run reached
|
| 59 |
-
20k steps, and it is the checkpoint the reported WD keyframe numbers come from.
|
| 60 |
-
It is the one place where the two conditions are not trained equally long.
|
| 61 |
-
|
| 62 |
-
The step directories these were renamed from, for cross-referencing against the
|
| 63 |
-
training logs: `050000` (the LeRobot trainer saves at `num_train_steps`,
|
| 64 |
-
zero-padded), `49999` and `20000` (openpi/orbax saves at
|
| 65 |
-
`num_train_steps - 1`, unpadded) and `model_epoch_1000.pth` (Robomimic is
|
| 66 |
-
epoch-based, 100 iterations per epoch).
|
| 67 |
|
| 68 |
UniSkill is a Robomimic diffusion policy over skill representations produced by
|
| 69 |
an **off-the-shelf, un-finetuned UniSkill inverse-skill-dynamics model**. That
|
|
|
|
| 33 |
|
| 34 |
Every method's weights are called **`ckpt`** whichever trainer wrote them — a
|
| 35 |
directory for the JAX/orbax and LeRobot policies, a single file for Vid2Robot
|
| 36 |
+
(`ckpt.pt`) and UniSkill (`ckpt.pth`).
|
|
|
|
| 37 |
|
| 38 |
`<condition>` is the training suite, and it is the **top** level: **`nd/`** = no
|
| 39 |
distraction (clean scenes), **`wd/`** = with distraction (distractors in both
|
|
|
|
| 46 |
|
| 47 |
## Checkpoints
|
| 48 |
|
| 49 |
+
| Paper label | Method dir | Weights | Size each |
|
| 50 |
+
|---|---|---|---|
|
| 51 |
+
| VLM + π0.5 | `vlm_pi05` | `ckpt/` | 9.35 GB |
|
| 52 |
+
| Vid. Cond. π0.5 (Uniform) | `vid_cond_pi05_uniform` | `ckpt/` | 6.20 GB |
|
| 53 |
+
| Vid. Cond. π0.5 (Keyframes) | `vid_cond_pi05_keyframes` | `ckpt/` | 6.20 GB |
|
| 54 |
+
| Vid2Robot | `vid2robot` | `ckpt.pt` | 2.37 GB |
|
| 55 |
+
| UniSkill | `uniskill_policy` | `ckpt.pth` | 0.74 GB |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
|
| 57 |
UniSkill is a Robomimic diffusion policy over skill representations produced by
|
| 58 |
an **off-the-shelf, un-finetuned UniSkill inverse-skill-dynamics model**. That
|