krish-simpler-tshirt-pi05-joint-delta-step2000-v1

djkesu/krish-simpler-tshirt-pi05-joint-delta-step2000-v1

OpenPI checkpoint exported from Modal training.

Checkpoint

• config: pi05_trlc
• experiment: krish_simpler_tshirt_pi05_joint_delta_50hz_bs16_2xa100_v1
• uploaded checkpoint step: 1999
• variant: custom
• includes train_state: true

Dataset

• dataset: Gongsta/krish-simpler-tshirt
• default prompt: "Pick up the t-shirt, stretch out the corners to lay it flat, and smooth out any wrinkles. Perform precise, sequential folds to condense it into a neat, perfectly aligned rectangle, matching the quality of a professional laundromat."

Files

• params/: inference-ready OpenPI parameters
• assets/: normalization stats required by OpenPI serving
• _CHECKPOINT_METADATA: checkpoint metadata
• checkpoint_info.json: upload metadata
• train_state/: resume state for further training

W&B

• run: krish_simpler_tshirt_pi05_joint_delta_50hz_bs16_2xa100_v1
• url: https://wandb.ai/km67000-palatial/openpi/runs/o4e3bpw5
• run val/best_loss: 0.029682738706469536
• run final step: 1900

Usage

Download the repo locally, then serve it with the OpenPI TRLC wrapper:

cd /workspace/openpi
uv run /workspace/recap/scripts/serve_trlc_openpi.py --variant custom --checkpoint-dir /path/to/downloaded/repo --port 8000