training_bundle/scripts/run_train_kinova.sh

#!/usr/bin/env bash
set -euo pipefail

export HF_HOME=/workspace/.cache/huggingface
export HF_HUB_CACHE="$HF_HOME/hub"
export HF_LEROBOT_HOME="$HF_HOME/lerobot"
export OPENPI_DATA_HOME=/workspace/.cache/openpi
export XDG_CACHE_HOME=/workspace/.cache
export TRANSFORMERS_CACHE="$HF_HOME/transformers"
export WANDB_DIR=/workspace/.cache/wandb
export TMPDIR=/workspace/.tmp
export PYTHONUNBUFFERED=1
export PYTHONFAULTHANDLER=1
export XLA_PYTHON_CLIENT_MEM_FRACTION=0.9

LOG_DIR=/workspace/logs
mkdir -p "$LOG_DIR"
LOG_FILE="$LOG_DIR/kinova_plush_box_v1_$(date +%Y%m%d_%H%M%S).log"

cd /workspace/openpi
/workspace/.venv/bin/python scripts/train.py pi05_kinova_plush_box --exp-name kinova_plush_box_v1 --overwrite 2>&1 | tee "$LOG_FILE"