blob: a97839341ad2f798aef4f588cc50aeaf656c67b7 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
#!/usr/bin/env bash
# Roadmap #3 (subgraph/ego features) + #4 (IGNN-style forced contraction), 5 seeds.
# #3: full --pe sub. #4: full --pe none --contract (vs free full/none baseline).
set -uo pipefail
cd /home/yurenh2/rrog
export PYTHONPATH=/home/yurenh2/rrog
echo "host=$(hostname) gpu=${CUDA_VISIBLE_DEVICES:-?} start=$(date -Is)"
for s in 0 1 2 3 4; do
ck=runs/ckpt_color_full_sub_n50_k3_p0.2_T3_ns3_s${s}.pt
echo "===== seed=$s #3 full sub ====="
python3 diag/train_color.py --mode train --grad_mode full --pe sub --p 0.2 --epochs 150 --seed "$s" || echo "!! train sub s$s failed"
python3 diag/train_color.py --mode le --ckpt "$ck" || echo "!! le sub s$s failed"
python3 diag/ptrm_color.py --ckpt "$ck" --K 16 --n_graphs 150 --sigmas 0.2 || echo "!! ptrm sub s$s failed"
ck2=runs/ckpt_color_full_none_ctr_n50_k3_p0.2_T3_ns3_s${s}.pt
echo "===== seed=$s #4 full none --contract ====="
python3 diag/train_color.py --mode train --grad_mode full --pe none --contract --p 0.2 --epochs 150 --seed "$s" || echo "!! train ctr s$s failed"
python3 diag/train_color.py --mode le --ckpt "$ck2" || echo "!! le ctr s$s failed"
python3 diag/ptrm_color.py --ckpt "$ck2" --K 16 --n_graphs 150 --sigmas 0.2 || echo "!! ptrm ctr s$s failed"
done
echo "===== AGGREGATE ====="
python3 diag/aggregate.py
echo "done=$(date -Is)"
|
