Robotics
LeRobot
Safetensors
pi05

plug_and_unplug_socket_no_vision

Pi0.5 policy fine-tuned with LeRobot.

Files

Standard LeRobot pi0.5 checkpoint layout:

File Purpose
model.safetensors Policy weights
config.json Policy config
policy_preprocessor.* Input normalization stats
policy_postprocessor.* Output unnormalization stats
train_config.json Training config snapshot

Loading

from lerobot.policies.pi05.modeling_pi05 import PI05Policy
policy = PI05Policy.from_pretrained("mcarroll101/plug_and_unplug_socket_no_vision")

(Manually pushed from /lus/lfs1aip2/projects/u6ed/milo/data/gim_arm/plug_and_unplug_socket_oneEP_May_6th_4gpu_frozen_vision_encoder_4/checkpoints/080000/pretrained_model via scripts/misc/push_pi05_checkpoint.py.)

Downloads last month
9
Safetensors
Model size
4B params
Tensor type
F32
·
BF16
·
Video Preview
loading

Dataset used to train mcarroll101/plug_and_unplug_socket_no_vision