Files
SIA/tools/train/bin/train
2025-03-18 09:46:16 +01:00

15 lines
308 B
Bash

#!/bin/bash
set -e
SIA_DIR="/root/sia"
OUTPUT_DIR="${1:-/root/models/$(cd "$SIA_DIR" && git rev-parse HEAD)}"
if [ -n "$(cd "$SIA_DIR" && git status --porcelain)" ]; then
echo "Uncommitted changes in SIA directory"
#exit 1
fi
mkdir -p "$OUTPUT_DIR"
/root/venvs/train/bin/python -m train.qwq "$@"