Robotics
LeRobot
Safetensors
pi05

Model Card for pi05

Ο€β‚€.β‚… (Pi05) Policy

This model which come from the Pytorch conversion script of openpi and their pi05_libero model, has been finetuned for 6k steps on 8x H100 GPU's.

Ο€β‚€.β‚… is a Vision-Language-Action model with open-world generalization, from Physical Intelligence. The LeRobot implementation is adapted from their open source OpenPI repository.

Model Overview

Ο€β‚€.β‚… represents a significant evolution from Ο€β‚€, developed by Physical Intelligence to address a big challenge in robotics: open-world generalization. While robots can perform impressive tasks in controlled environments, Ο€β‚€.β‚… is designed to generalize to entirely new environments and situations that were never seen during training.

For more details, see the Physical Intelligence Ο€β‚€.β‚… blog post.

This policy has been trained and pushed to the Hub using LeRobot. See the full documentation at LeRobot Docs.


How to Get Started with the Model

For a complete walkthrough, see the training guide. Below is the short version on how to train and run inference/eval:

python src/lerobot/scripts/train.py \
    --dataset.repo_id=your_dataset \
    --policy.type=pi05 \
    --output_dir=./outputs/pi05_training \
    --job_name=pi05_training \
    --policy.repo_id=your_repo_id \
    --policy.pretrained_path=lerobot/pi05_libero_finetuned \
    --policy.compile_model=true \
    --policy.gradient_checkpointing=true \
    --wandb.enable=true \
    --policy.dtype=bfloat16 \
    --steps=3000 \
    --policy.scheduler_decay_steps=3000 \
    --policy.device=cuda \
    --batch_size=32

Model Details

  • License: apache-2.0
Downloads last month
1,741
Safetensors
Model size
4B params
Tensor type
F32
Β·
BF16
Β·
Video Preview
loading

Dataset used to train lerobot/pi05_libero_finetuned

Collection including lerobot/pi05_libero_finetuned