haohw commited on
Commit
c4073c2
·
verified ·
1 Parent(s): bd16683

add README.md

Browse files
Files changed (1) hide show
  1. README.md +30 -0
README.md ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ library_name: pytorch
4
+ ---
5
+
6
+ # Push-T diffusion policy checkpoints
7
+
8
+ Two Push-T diffusion policy checkpoints with a minimal inference script.
9
+
10
+ ## Files
11
+
12
+ - `original.ckpt` — original checkpoint
13
+ - `edited.pt` — edited checkpoint
14
+ - `stats_sfp.npz` — normalization stats (obs / action min/max)
15
+ - `inference.py` — inference script
16
+
17
+ ## Quick start
18
+
19
+ ```bash
20
+ pip install torch diffusers pygame pymunk shapely scikit-image imageio gym numpy
21
+ git clone https://github.com/columbia-ai-robotics/streaming_flow_policy.git
22
+
23
+ python inference.py --ckpt original.ckpt --stats stats_sfp.npz \
24
+ --sfp-repo streaming_flow_policy/ --n-seeds 50
25
+
26
+ python inference.py --ckpt edited.pt --stats stats_sfp.npz \
27
+ --sfp-repo streaming_flow_policy/ --n-seeds 50 --save-mp4 rollout.mp4
28
+ ```
29
+
30
+ Prints success rate and mode distribution; optional `--save-mp4` renders the first successful rollout.