#!/bin/bash
# RunPod Training Script for Bamboo-1
# Usage: bash scripts/runpod_train.sh

set -e

echo "=========================================="
echo "Bamboo-1: Vietnamese Dependency Parser"
echo "RunPod Training Setup"
echo "=========================================="

# Install uv if not present
if ! command -v uv &> /dev/null; then
    echo "Installing uv..."
    curl -LsSf https://astral.sh/uv/install.sh | sh
    source $HOME/.local/bin/env
fi

# Clone repo if not exists
if [ ! -d "bamboo-1" ]; then
    echo "Cloning bamboo-1 from HuggingFace..."
    git clone https://huggingface.co/undertheseanlp/bamboo-1
fi

cd bamboo-1

# Install dependencies
echo "Installing dependencies..."
uv sync

# Run training
echo "Starting training..."
uv run scripts/train.py \
    --output models/bamboo-1-char \
    --feat char \
    --max-epochs 100 \
    --batch-size 5000 \
    --lr 2e-3 \
    "$@"

echo "Training complete!"
echo "Model saved to: models/bamboo-1-char"