diff options
Diffstat (limited to 'diag/run_pe3.sh')
| -rw-r--r-- | diag/run_pe3.sh | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/diag/run_pe3.sh b/diag/run_pe3.sh index a978393..2a6744b 100644 --- a/diag/run_pe3.sh +++ b/diag/run_pe3.sh @@ -1,22 +1,20 @@ #!/usr/bin/env bash -# Roadmap #3 (subgraph/ego features) + #4 (IGNN-style forced contraction), 5 seeds. +# Subgraph/ego features + forced contraction, 5 seeds. Deterministic train + LE. # #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 + ck=runs/ckpt_color_rrog_trm_gin_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 + ck2=runs/ckpt_color_rrog_trm_gin_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 |