Views
No views yet
lerobot/pi0_base on
collected-ai/lehome-four-types-merged (dual-arm SO101 garment folding; 3 RGB cameras 480x640 +
12-dim joint state; task "fold the garment on the table").main is the final step-30000 model; intermediates are branches:
step-5000, step-10000, step-15000, step-20000, step-25000.1from lerobot.policies.pi0.modeling_pi0 import PI0Policy
2policy = PI0Policy.from_pretrained("collected-ai/pi0_lehome_four_types") # step 30000
3policy = PI0Policy.from_pretrained("collected-ai/pi0_lehome_four_types", revision="step-15000")