glory-hyeok/pi0-fast-bridge-simpler-horizon-5-bs64-s-0.05-10k

OpenPI JAX Checkpoint

This repository contains a complete JAX checkpoint from OpenPI training.

Model Details:

• Framework: JAX/Flax
• Original path: /virtual_lab/sjw_alinlab/suhyeok/openpi/checkpoints/pi0_fast_bridge_simpler_base_horizon_5/pi0-fast-bridge-simpler-base-horizon-5-bs64-state-drop-0.05/10000
• Checkpoint step: 10000

Files

This checkpoint contains all original files from the training run:

• params: Model parameters (JAX format)
• opt_state: Optimizer state
• metadata.json: Training metadata (if available)
• Other checkpoint files

Loading

To load this checkpoint in OpenPI:

import openpi.training.checkpoints as checkpoints

# Load the checkpoint
checkpoint_dir = "path/to/downloaded/checkpoint"
# Use OpenPI's checkpoint loading utilities

Usage

This checkpoint can be used directly with OpenPI's evaluation and inference scripts.