Files
SIA/scripts/train.sh
2025-05-20 20:47:57 +02:00

17 lines
364 B
Bash

#!/bin/bash
set -e
SIA_DIR="/root/sia"
OUTPUT_DIR="${1:-/root/models/$(cd "$SIA_DIR" && git rev-parse HEAD)}"
echo "Output dir: $OUTPUT_DIR"
if [ -n "$(cd "$SIA_DIR" && git status --porcelain)" ]; then
echo "Uncommitted changes in SIA directory"
#exit 1
fi
mkdir -p "$OUTPUT_DIR"
train_gemma --model "google/gemma-3-12b-it" --output-dir "$OUTPUT_DIR"