LogosRoboticsLab commited on
Commit
ef21150
·
verified ·
1 Parent(s): a3f6333

Drop the training-step numbers from the model card

Browse files
Files changed (1) hide show
  1. README.md +8 -19
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`). The training step each one corresponds
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 | Step — `nd/` | Step — `wd/` | Size each |
51
- |---|---|---|---|---|---|
52
- | VLM + π0.5 | `vlm_pi05` | `ckpt/` | 50,000 | 50,000 | 9.35 GB |
53
- | Vid. Cond. π0.5 (Uniform) | `vid_cond_pi05_uniform` | `ckpt/` | 50,000 | 50,000 | 6.20 GB |
54
- | Vid. Cond. π0.5 (Keyframes) | `vid_cond_pi05_keyframes` | `ckpt/` | 50,000 | **20,000** | 6.20 GB |
55
- | Vid2Robot | `vid2robot` | `ckpt.pt` | 50,000 | 50,000 | 2.37 GB |
56
- | UniSkill | `uniskill_policy` | `ckpt.pth` | epoch 1,000 | epoch 1,000 | 0.74 GB |
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