Diffusion Policy: Visuomotor Policy Learning via Action Diffusion
Paper • 2303.04137 • Published • 6
How to use DreamMachines/actuator_unboxing_20h_345_dp with LeRobot:
Diffusion Policy — visuomotor control as a generative diffusion process. See the paper.
Trained with LeDream (a fork of LeRobot).
DreamMachines/actuator_unboxing_20h_34520000064200000adam (lr 0.0001)Every training checkpoint is uploaded as its own git revision; the latest is also on main. Load a specific step with:
from lerobot.policies.factory import make_policy
from lerobot.configs.policies import PreTrainedConfig
cfg = PreTrainedConfig.from_pretrained("DreamMachines/actuator_unboxing_20h_345_dp", revision="v002000")
policy = make_policy(cfg)
Available: v020000, v040000, v060000, v080000, v100000, v120000, v140000, v160000, v180000, v200000
make eval MODEL=DreamMachines/actuator_unboxing_20h_345_dp
See docs/source/policy_pi05_README.md for π₀.₅-specific notes.