ExpoFT-Pi05-StackCube-v1-SFT

ExpoFT π₀.₅ — StackCube-v1 (SFT)

LoRA-finetuned checkpoint of π₀.₅ (pi05_droid_jointpos), adapted to ManiSkill's StackCube-v1 task as part of the ExpoFT reproduction project.

• Success rate: 41% on 200 held-out fixed-seed evaluation episodes (SE = sqrt(p(1-p)/n))
• Training: LoRA supervised fine-tuning on 50 motion-planning demonstrations
• Action space: joint-space (pd_joint_delta_pos), matching pi0.5's DROID pretraining distribution
• Normalization stats: official DROID stats (gs://openpi-assets/checkpoints/pi05_droid_jointpos/assets/droid)

Warning: JAX only

This checkpoint is in its native JAX/orbax format (as produced by openpi training) and has not been converted to PyTorch. openpi's official conversion script (examples/convert_jax_model_to_pytorch.py) exists, but PyTorch-side LoRA support is currently absent in openpi, and unresolved upstream issues report output mismatches between JAX and converted PyTorch weights for pi0.5 specifically -- so no PyTorch conversion is provided here.

Usage

Point pi05_weight_loader_path at the downloaded params/ directory when building the model via openpi's CheckpointWeightLoader, using the expo_pi05_droid_lora_finetune_sft_joint_state training config as the base.

Code

Training/eval pipeline: Josh012006/expo-ft