Upload folder using huggingface_hub
Browse files- README.md +47 -0
- eval_records.tgz +3 -0
- pushf_finetune_n100_seed0.pt +3 -0
- pushf_scratch_n100_seed0.pt +3 -0
- pusht_1step_distilled_seed0.pt +3 -0
- pusht_2rf_seed0.pt +3 -0
- pusht_ddpm_unet_seed0.pt +3 -0
- pusht_fm_dit_seed0.pt +3 -0
- pusht_fm_unet_image_seed0.pt +3 -0
- pusht_fm_unet_seed0.pt +3 -0
README.md
CHANGED
|
@@ -1,3 +1,50 @@
|
|
| 1 |
---
|
| 2 |
license: mit
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
+
library_name: pytorch
|
| 4 |
+
tags:
|
| 5 |
+
- robotics
|
| 6 |
+
- imitation-learning
|
| 7 |
+
- diffusion-policy
|
| 8 |
+
- flow-matching
|
| 9 |
+
- push-t
|
| 10 |
+
datasets:
|
| 11 |
+
- WilliamHangXu/pushf-demos
|
| 12 |
---
|
| 13 |
+
|
| 14 |
+
# FlowPush checkpoints
|
| 15 |
+
|
| 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; evaluating
|
| 20 |
+
it on the 50 test episodes reproduces that seed's number (the README reports 3-seed means).
|
| 21 |
+
|
| 22 |
+
| file | model | success@0.95, seed 0 |
|
| 23 |
+
|---|---|---|
|
| 24 |
+
| `pusht_ddpm_unet_seed0.pt` | DDPM U-Net, state obs, 100 steps | 0.44 |
|
| 25 |
+
| `pusht_fm_unet_seed0.pt` | flow-matching U-Net, state obs, 8 Euler steps (the teacher) | 0.60 |
|
| 26 |
+
| `pusht_fm_dit_seed0.pt` | flow-matching DiT (7.8M params), state obs, 8 steps | 0.34 |
|
| 27 |
+
| `pusht_fm_unet_image_seed0.pt` | flow-matching U-Net, 96×96 RGB obs, 8 steps | 0.30 |
|
| 28 |
+
| `pusht_2rf_seed0.pt` | 2-rectified flow, 1 step | 0.46 |
|
| 29 |
+
| `pusht_1step_distilled_seed0.pt` | 1-step distilled student | 0.52 |
|
| 30 |
+
| `pushf_scratch_n100_seed0.pt` | Push-F, trained from scratch on 100 demos, 8 steps | 0.66 |
|
| 31 |
+
| `pushf_finetune_n100_seed0.pt` | Push-F, fine-tuned from the Push-T teacher on 100 demos (lr 1e-4), 8 steps | 0.78 |
|
| 32 |
+
|
| 33 |
+
`eval_records.tgz` holds the per-episode evaluation records behind the repository's
|
| 34 |
+
`results/*.csv`.
|
| 35 |
+
|
| 36 |
+
## Usage
|
| 37 |
+
|
| 38 |
+
```bash
|
| 39 |
+
git clone https://github.com/WilliamHangXu/FlowPush.git && cd FlowPush
|
| 40 |
+
uv sync && make data && make checkpoints
|
| 41 |
+
uv run python evaluate.py --ckpt checkpoints/pusht_fm_unet_seed0.pt --seeds test --points euler:8
|
| 42 |
+
# Push-F policies: add --env pushf
|
| 43 |
+
```
|
| 44 |
+
|
| 45 |
+
Samplers and step counts are evaluation-time choices (`--points euler:1`, `ddim:16`, …).
|
| 46 |
+
Checkpoints are self-contained: no side files are needed.
|
| 47 |
+
|
| 48 |
+
## License
|
| 49 |
+
|
| 50 |
+
MIT.
|
eval_records.tgz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:15887bd52b7f674eb6ba349f63d6eb0442e9f84fe78731fd77f864e94a884040
|
| 3 |
+
size 140659
|
pushf_finetune_n100_seed0.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dfbec70a3bd8433e8d27e962517096614a004e537c1705d47ca33c697536e88f
|
| 3 |
+
size 261596075
|
pushf_scratch_n100_seed0.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0115959f5a0a4a64cdb2cc907f410496817c361b60974e9b07c83104a1bebd14
|
| 3 |
+
size 261595853
|
pusht_1step_distilled_seed0.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d95bafa0f2917fffaf800c712c5894d9d998a97124802b5b821b1bc6c348c6c8
|
| 3 |
+
size 261596327
|
pusht_2rf_seed0.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f7a7323c439351208625cead9d6a0b5314eb58aaac9d405c7cf0d42befb58672
|
| 3 |
+
size 261594367
|
pusht_ddpm_unet_seed0.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:32f69332dc31efbd63b03f82c738601a34778505a1e85bc5aaf21fa8e3b76bc5
|
| 3 |
+
size 261595251
|
pusht_fm_dit_seed0.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7d4538b8403a9f8b06d7be3737cea79fa852e493ee008629758b7426aa1f9355
|
| 3 |
+
size 31098957
|
pusht_fm_unet_image_seed0.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:85cae1f7b85a1999edb0cae5130a9247bad0736d4d4d947930f0cc3932805d8b
|
| 3 |
+
size 1050911691
|
pusht_fm_unet_seed0.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:86fdd4af3960143230195b01d169b32929600080a79a1274a1f16a81c17dc388
|
| 3 |
+
size 261594871
|