#!/usr/bin/env bash
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
RECIPE="$(cd "$SCRIPT_DIR/.." && pwd)"
GPU_L2A="${GPU_L2A:-0}"
cd "$RECIPE"

echo "L2A-style = per-token, head-shared gate; assigned visible GPU $GPU_L2A."
echo "Training L2A-style variant only."

python "$SCRIPT_DIR/validate_release.py"
python -m unittest -v tests.test_router_granularity

"$SCRIPT_DIR/train_one.sh" l2a_style "$GPU_L2A"

echo "L2A-style selected checkpoint: ${OUTPUT_ROOT:-$RECIPE/../outputs}/l2a_style/stage2/checkpoint-25"