#!/bin/bash
# Grid 12 - inflection across typologically diverse languages.
# Reported as the BOUNDARY: answers are unique but shallow, so one pass already
# works and no method is needed. Breadth here is across languages, not seeds.
#SBATCH -J g12
#SBATCH -p gpu-h100
#SBATCH --gpus=h100:1
#SBATCH -c 8
#SBATCH --mem=32g
#SBATCH -t 06:00:00
#SBATCH -o logs/g12-%A_%a.out
#SBATCH --array=0-23%12
source /etc/profile
SCR=/scratch/zt1/project/msml612/user/$USER
module load python/3.10.10
export PYTHONPATH=
cd $SCR/pcmd/src
LANGS=(english german spanish turkish finnish russian navajo georgian)
MODES=(mdlm full); SEEDS=(0)
I=$SLURM_ARRAY_TASK_ID
L=${LANGS[$((I / 3))]}
M=${MODES[$(((I % 3) / 2))]}
NAME="g12-$L-$M"
echo "=== $NAME ==="
$SCR/hf_env/bin/python train_sig.py --lang $L --mode $M --size medium --seed 0 \
  --steps 15000 --d 256 --layers 6 --lr 1e-4 --warmup 1000 \
  --data_root $SCR/pcmd/data/sig --out $SCR/pcmd/runs/$NAME