blob: 5385152b3ff2738771b98ff6203376d5ca37b404 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
#!/usr/bin/env bash
# Conv axis (pe=none): gcn, sage, gat. 5 seeds, train+LE+PTRM. (pin GPU at launch.)
set -uo pipefail
cd /home/yurenh2/rrog
export PYTHONPATH=/home/yurenh2/rrog
echo "A start gpu=${CUDA_VISIBLE_DEVICES:-?} $(date -Is)"
for s in 0 1 2 3 4; do
for conv in gcn sage gat; do
ck=runs/ckpt_color_${conv}_full_none_n50_k3_p0.2_T3_ns3_s${s}.pt
echo "== A s$s conv=$conv =="
python3 diag/train_color.py --mode train --conv "$conv" --pe none --p 0.2 --epochs 150 --seed "$s" || echo "!! train $conv s$s"
python3 diag/train_color.py --mode le --ckpt "$ck" || echo "!! le $conv s$s"
python3 diag/ptrm_color.py --ckpt "$ck" --K 16 --n_graphs 150 --sigmas 0.2 || echo "!! ptrm $conv s$s"
done
done
echo "doneA=$(date -Is)"
|
