Gemma training script

scripts/train.sh

@@ -0,0 +1,17 @@
+#!/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 --output-dir "$OUTPUT_DIR"