WillXH commited on
Commit
94e7ea4
·
verified ·
1 Parent(s): a451e4c

Upload folder using huggingface_hub

Browse files
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