#!/bin/bash
#SBATCH --job-name=qcpinn_cavity_public
#SBATCH --cpus-per-task=4
#SBATCH --mem=16G
#SBATCH --time=16:00:00
set -euo pipefail
cd "${SLURM_SUBMIT_DIR:?Submit from the model package root}"
export OMP_NUM_THREADS=1
export OPENBLAS_NUM_THREADS=1
export MKL_NUM_THREADS=1
export MPLBACKEND=Agg

pids=()
for seed in {0..9}; do
  (
    python -u scripts/train.py --config config/config.yaml --seed "$seed"
    python -u scripts/inference.py --config config/config.yaml --seed "$seed" --kind best
    python -u scripts/inference.py --config config/config.yaml --seed "$seed" --kind last --metrics-only
    python -u scripts/result.py --config config/config.yaml --seed "$seed"
  ) &
  pids+=("$!")
  if ((${#pids[@]} == 4)); then
    for pid in "${pids[@]}"; do wait "$pid"; done
    pids=()
  fi
done
for pid in "${pids[@]}"; do wait "$pid"; done
python -u scripts/result.py --config config/config.yaml --aggregate
python -u scripts/inference.py --config config/config.yaml
python -u scripts/result.py --config config/config.yaml