Downloads · 30 days
0
pusht-dp-checkpoints
haohw/pusht-dp-checkpoints
pusht-dp-checkpoints is a machine learning model from haohw. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for pytorch. The card lists the license as mit.
Two Push-T diffusion policy checkpoints with a minimal inference script.
Access
Public
Updated Jun 5, 2026
Repo size
523 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt261 MB · 50%
At a glance
- Library
- pytorch
- License
- mit
- Access
- Public
- Created
- Jun 5, 2026
- Updated
- Jun 5, 2026
- SHA
- c4073c29
Model card
From the Hugging Face model README
Push-T diffusion policy checkpoints
Two Push-T diffusion policy checkpoints with a minimal inference script.
Files
original.ckpt— original checkpointedited.pt— edited checkpointstats_sfp.npz— normalization stats (obs / action min/max)inference.py— inference script
Quick start
pip install torch diffusers pygame pymunk shapely scikit-image imageio gym numpy
git clone https://github.com/columbia-ai-robotics/streaming_flow_policy.git
python inference.py --ckpt original.ckpt --stats stats_sfp.npz \
--sfp-repo streaming_flow_policy/ --n-seeds 50
python inference.py --ckpt edited.pt --stats stats_sfp.npz \
--sfp-repo streaming_flow_policy/ --n-seeds 50 --save-mp4 rollout.mp4
Prints success rate and mode distribution; optional --save-mp4 renders the first successful rollout.
- Library
- pytorch
- License
- mit
- Created
- Jun 5, 2026
- Updated
- Jun 5, 2026