blob: 4a6f31afe0219b258139346f1425deac94ed86ba (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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)"
|
