code/scripts/run_sanity_check.sh

#!/usr/bin/env bash
# scripts/run_sanity_check.sh
# Fires both the standard control AND the full grokking recipe, in parallel,
# each detached under nohup. Use this as the Week 2 go/no-go signal.

set -euo pipefail

ROOT="$(cd "$(dirname "$0")/.." && pwd)"
cd "${ROOT}"

GPU_STD="${GPU_STD:-0}"
GPU_GROK="${GPU_GROK:-1}"
N_TRAIN="${N_TRAIN:-500}"
SEED="${SEED:-42}"

echo "Sanity check — both conditions, detached"
bash scripts/launch.sh standard "${N_TRAIN}" "${SEED}" "${GPU_STD}"
bash scripts/launch.sh grokking "${N_TRAIN}" "${SEED}" "${GPU_GROK}"

echo
echo "Both runs detached. Watch progress with:"
echo "  bash scripts/list_runs.sh"
echo "  tail -f experiments/runs/<run_id>/logs/train.log"