diff options
Diffstat (limited to 'diag/run_diag.sh')
| -rw-r--r-- | diag/run_diag.sh | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/diag/run_diag.sh b/diag/run_diag.sh new file mode 100644 index 0000000..4a6f31a --- /dev/null +++ b/diag/run_diag.sh @@ -0,0 +1,16 @@ +#!/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)" |