56 lines
1.8 KiB
PowerShell
56 lines
1.8 KiB
PowerShell
# NNUE Pipeline launcher for PowerShell (Windows)
|
|
|
|
Write-Host "Launching NNUE Training Pipeline..." -ForegroundColor Green
|
|
Write-Host ""
|
|
|
|
# Check if we're in the right directory
|
|
if (!(Test-Path "python")) {
|
|
Write-Host "ERROR: python directory not found" -ForegroundColor Red
|
|
Write-Host "Please run this script from the modules\bot directory" -ForegroundColor Red
|
|
exit 1
|
|
}
|
|
|
|
# Check for Stockfish
|
|
$stockfishPath = $env:STOCKFISH_PATH
|
|
if ($null -eq $stockfishPath -or $stockfishPath -eq "") {
|
|
Write-Host "Stockfish path not set. Trying to find in PATH..." -ForegroundColor Yellow
|
|
$stockfishPath = (Get-Command stockfish -ErrorAction SilentlyContinue).Source
|
|
if ($null -eq $stockfishPath) {
|
|
Write-Host "Stockfish not found in PATH" -ForegroundColor Yellow
|
|
Write-Host "Set STOCKFISH_PATH environment variable and try again:" -ForegroundColor Yellow
|
|
Write-Host ' $env:STOCKFISH_PATH = "C:\path\to\stockfish.exe"' -ForegroundColor Cyan
|
|
} else {
|
|
Write-Host "Found Stockfish: $stockfishPath" -ForegroundColor Green
|
|
$env:STOCKFISH_PATH = $stockfishPath
|
|
}
|
|
}
|
|
|
|
# Run the pipeline
|
|
Write-Host "Running pipeline from: $(Get-Location)\python" -ForegroundColor Cyan
|
|
Write-Host ""
|
|
|
|
Push-Location python
|
|
try {
|
|
# Use bash if available (Git Bash or WSL)
|
|
if (Get-Command bash -ErrorAction SilentlyContinue) {
|
|
Write-Host "Using bash script..." -ForegroundColor Cyan
|
|
bash ./run_pipeline.sh
|
|
} else {
|
|
Write-Host "Using batch script..." -ForegroundColor Cyan
|
|
& cmd.exe /c run_pipeline.bat
|
|
}
|
|
$result = $LASTEXITCODE
|
|
} finally {
|
|
Pop-Location
|
|
}
|
|
|
|
if ($result -eq 0) {
|
|
Write-Host ""
|
|
Write-Host "Pipeline completed successfully!" -ForegroundColor Green
|
|
} else {
|
|
Write-Host ""
|
|
Write-Host "Pipeline failed with exit code $result" -ForegroundColor Red
|
|
}
|
|
|
|
exit $result
|