#!/usr/bin/env bash
# Step-1 diagnosis sweep. CSL = pure-H2 anchor (classification).
# Triangle counting on ER (graphs 1-WL-distinguishable -> H2~0, the H1 end) and
# regular (graphs collapse to one WL color -> H2~var, the ceiling end).
set -uo pipefail
cd /home/yurenh2/rrog
export PYTHONPATH=/home/yurenh2/rrog
echo "host=$(hostname) gpu=${CUDA_VISIBLE_DEVICES:-?} start=$(date -Is)"
run() { echo "===== $* ====="; python3 diag/train_diag.py "$@" --layers 4 --epochs 300 --seed 0 \
        || echo "!! FAILED $*"; }
for model in gin gcn; do
  run --task csl --model "$model"
  run --task tri --model "$model" --kind er
  run --task tri --model "$model" --kind regular
done
echo "done=$(date -Is)"