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

variant="${1:-primary}"
device="${2:-cuda}"
mkdir -p results/oral_a_dev

common=(
  --mode bp --device "${device}" --depth 20 --width 16 --seed 0 --loader_seed 0
  --batch_size 128 --epochs 200 --val_examples 5000 --split_seed 2027
  --eval_split validation --eval_every 20 --augment_train 1
  --lr 0.1 --momentum 0.9 --normalization batchnorm
)

case "${variant}" in
  primary)
    python3 experiments/conv_run.py "${common[@]}" \
      --lr_schedule step --lr_milestones 100,150 --lr_gamma 0.1 \
      --warmup_epochs 0 --weight_decay 1e-4 \
      --out results/oral_a_dev/bp_reference_primary.json
    ;;
  recovery)
    python3 experiments/conv_run.py "${common[@]}" \
      --lr_schedule cosine --warmup_epochs 5 --weight_decay 5e-4 \
      --out results/oral_a_dev/bp_reference_recovery.json
    ;;
  *)
    echo "variant must be primary or recovery" >&2
    exit 2
    ;;
esac