Upload folder using huggingface_hub
Browse files- README.md +20 -16
- pusht_1step_distilled_seed0.pt +2 -2
- pusht_2rf_k32_seed0.pt +2 -2
- pusht_2rf_seed0.pt +2 -2
README.md
CHANGED
|
@@ -16,27 +16,31 @@ datasets:
|
|
| 16 |
Evaluation-only checkpoints (EMA weights, fitted normalizer, resolved config, git SHA) for
|
| 17 |
[FlowPush](https://github.com/WilliamHangXu/FlowPush): flow-matching and diffusion
|
| 18 |
visuomotor policies on Push-T, a 1-step distilled student, and Push-F transfer policies.
|
| 19 |
-
Each file is training seed 0 of the corresponding row in the repository README
|
| 20 |
-
|
|
|
|
|
|
|
| 21 |
|
| 22 |
-
| file | model | success@0.95
|
| 23 |
|---|---|---|
|
| 24 |
-
| `pusht_ddpm_unet_seed0.pt` | DDPM U-Net, state obs, 100 steps | 0.
|
| 25 |
-
| `pusht_fm_unet_seed0.pt` | flow-matching U-Net, state obs, 8 Euler steps | 0.
|
| 26 |
-
| `pusht_fm_unet_seed2.pt` | flow-matching U-Net, state obs — the distillation teacher | 0.
|
| 27 |
-
| `pusht_fm_dit_seed0.pt` | flow-matching DiT (7.8M params), state obs, 8 steps | 0.34 |
|
| 28 |
-
| `pusht_fm_unet_image_seed0.pt` | flow-matching U-Net, 96×96 RGB obs, 8 steps | 0.30 |
|
| 29 |
-
| `pusht_2rf_seed0.pt` | 2-rectified flow (K = 8), 1 step | 0.
|
| 30 |
-
| `pusht_2rf_k32_seed0.pt` | 2-rectified flow (K = 32) — parent of the 1-step students | 0.
|
| 31 |
-
| `pusht_1step_distilled_seed0.pt` | 1-step distilled student (trained from `pusht_2rf_k32_seed0.pt`) | 0.
|
| 32 |
-
| `pushf_scratch_n100_seed0.pt` | Push-F, trained from scratch on 100 demos, 8 steps | 0.
|
| 33 |
-
| `pushf_finetune_n100_seed0.pt` | Push-F, fine-tuned from the Push-T teacher on 100 demos (lr 1e-4), 8 steps | 0.
|
| 34 |
|
| 35 |
`eval_records.tgz` holds the per-episode evaluation records behind the repository's
|
| 36 |
`results/*.csv`. All three student seeds share the single K = 32 parent above, which was
|
| 37 |
trained from the seed-2 teacher; the repository's `results/lineage.csv` records parents and
|
| 38 |
-
SHA-256 hashes
|
| 39 |
-
|
|
|
|
|
|
|
| 40 |
(100 fresh episodes, one evaluation per model; `docs/FINAL_HOLDOUT.md` in the repository)
|
| 41 |
re-scored every released Push-T/Push-F policy: FM@8 0.617 ± 0.031, DDPM@100 0.537 ± 0.127,
|
| 42 |
1-step student 0.427 ± 0.025 vs its K = 32 parent 0.470 — the student's development-set
|
|
@@ -48,7 +52,7 @@ of record; `eval_records.tgz` includes the `final_*.json` records.
|
|
| 48 |
```bash
|
| 49 |
git clone https://github.com/WilliamHangXu/FlowPush.git && cd FlowPush
|
| 50 |
uv sync && make data && make checkpoints
|
| 51 |
-
uv run python evaluate.py --ckpt checkpoints/pusht_fm_unet_seed0.pt --seeds
|
| 52 |
# Push-F policies: add --env pushf
|
| 53 |
```
|
| 54 |
|
|
|
|
| 16 |
Evaluation-only checkpoints (EMA weights, fitted normalizer, resolved config, git SHA) for
|
| 17 |
[FlowPush](https://github.com/WilliamHangXu/FlowPush): flow-matching and diffusion
|
| 18 |
visuomotor policies on Push-T, a 1-step distilled student, and Push-F transfer policies.
|
| 19 |
+
Each file is training seed 0 of the corresponding row in the repository README (plus the
|
| 20 |
+
seed-2 teacher); scores are that seed's result on the 100-episode final holdout — † marks
|
| 21 |
+
models outside the holdout list, scored on the 50-episode development set. The README
|
| 22 |
+
reports 3-seed means.
|
| 23 |
|
| 24 |
+
| file | model | success@0.95 (final holdout) |
|
| 25 |
|---|---|---|
|
| 26 |
+
| `pusht_ddpm_unet_seed0.pt` | DDPM U-Net, state obs, 100 steps | 0.62 |
|
| 27 |
+
| `pusht_fm_unet_seed0.pt` | flow-matching U-Net, state obs, 8 Euler steps | 0.65 |
|
| 28 |
+
| `pusht_fm_unet_seed2.pt` | flow-matching U-Net, state obs — the distillation teacher | 0.61 |
|
| 29 |
+
| `pusht_fm_dit_seed0.pt` | flow-matching DiT (7.8M params), state obs, 8 steps | 0.34† |
|
| 30 |
+
| `pusht_fm_unet_image_seed0.pt` | flow-matching U-Net, 96×96 RGB obs, 8 steps | 0.30† |
|
| 31 |
+
| `pusht_2rf_seed0.pt` | 2-rectified flow (K = 8), 1 step | 0.47 |
|
| 32 |
+
| `pusht_2rf_k32_seed0.pt` | 2-rectified flow (K = 32) — parent of the 1-step students | 0.47 |
|
| 33 |
+
| `pusht_1step_distilled_seed0.pt` | 1-step distilled student (trained from `pusht_2rf_k32_seed0.pt`) | 0.40 |
|
| 34 |
+
| `pushf_scratch_n100_seed0.pt` | Push-F, trained from scratch on 100 demos, 8 steps | 0.70 |
|
| 35 |
+
| `pushf_finetune_n100_seed0.pt` | Push-F, fine-tuned from the Push-T teacher on 100 demos (lr 1e-4), 8 steps | 0.67 |
|
| 36 |
|
| 37 |
`eval_records.tgz` holds the per-episode evaluation records behind the repository's
|
| 38 |
`results/*.csv`. All three student seeds share the single K = 32 parent above, which was
|
| 39 |
trained from the seed-2 teacher; the repository's `results/lineage.csv` records parents and
|
| 40 |
+
SHA-256 hashes, and the 2-RF and student checkpoints embed their coupling manifest
|
| 41 |
+
(teacher SHA-256, K, ODE steps). The original 50-episode evaluation set was consulted during
|
| 42 |
+
exploratory work in Parts 2 and 3, so numbers from it are development results (marked †).
|
| 43 |
+
A final holdout, pre-specified and committed before evaluation
|
| 44 |
(100 fresh episodes, one evaluation per model; `docs/FINAL_HOLDOUT.md` in the repository)
|
| 45 |
re-scored every released Push-T/Push-F policy: FM@8 0.617 ± 0.031, DDPM@100 0.537 ± 0.127,
|
| 46 |
1-step student 0.427 ± 0.025 vs its K = 32 parent 0.470 — the student's development-set
|
|
|
|
| 52 |
```bash
|
| 53 |
git clone https://github.com/WilliamHangXu/FlowPush.git && cd FlowPush
|
| 54 |
uv sync && make data && make checkpoints
|
| 55 |
+
uv run python evaluate.py --ckpt checkpoints/pusht_fm_unet_seed0.pt --seeds final --points euler:8
|
| 56 |
# Push-F policies: add --env pushf
|
| 57 |
```
|
| 58 |
|
pusht_1step_distilled_seed0.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f9d9243d54e333e90ce9afaf6c61b8c6378a078e9546829ece2ee5b27d49f347
|
| 3 |
+
size 261597095
|
pusht_2rf_k32_seed0.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a57dd49dc6d68b7423b95bc627812496979b4cabdbfc17bfd12940efad1eed9f
|
| 3 |
+
size 261595767
|
pusht_2rf_seed0.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8634a74d93899ea9a60b043a3a6865810d85b35602a978f0e8984492bb0a50bb
|
| 3 |
+
size 261595135
|