30 lines
817 B
Bash
30 lines
817 B
Bash
#!/bin/bash
|
|
# Setup and run NNUE training pipeline
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
VENV_DIR="$SCRIPT_DIR/.venv"
|
|
|
|
# Check if virtual environment exists
|
|
if [ ! -d "$VENV_DIR" ]; then
|
|
echo "Creating virtual environment..."
|
|
python3 -m venv "$VENV_DIR"
|
|
if [ $? -ne 0 ]; then
|
|
echo "Error: Failed to create virtual environment. Make sure python3 is installed."
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
# Activate virtual environment
|
|
echo "Activating virtual environment..."
|
|
source "$VENV_DIR/bin/activate"
|
|
|
|
# Install/update dependencies if requirements.txt exists
|
|
if [ -f "$SCRIPT_DIR/requirements.txt" ]; then
|
|
echo "Installing dependencies..."
|
|
pip install -q -r "$SCRIPT_DIR/requirements.txt"
|
|
fi
|
|
|
|
# Run nnue.py
|
|
echo "Starting NNUE Training Pipeline..."
|
|
python "$SCRIPT_DIR/nnue.py"
|