diff options
Diffstat (limited to 'diag/run_pe2.sh')
| -rw-r--r-- | diag/run_pe2.sh | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/diag/run_pe2.sh b/diag/run_pe2.sh index db7bd8a..dc9f88b 100644 --- a/diag/run_pe2.sh +++ b/diag/run_pe2.sh @@ -1,17 +1,15 @@ #!/usr/bin/env bash -# Roadmap #2: RRoG on a GSN-style motif backbone (per-node K3/wedge substructure counts). -# full-recursion, 5 seeds; train + LE + PTRM(sigma=0.2). Aggregate vs none/rwse. +# RRoG on a GSN-style motif-augmented one-shot encoder. Full recursion, 5 seeds; train + LE. 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_gsn_n50_k3_p0.2_T3_ns3_s${s}.pt + ck=runs/ckpt_color_rrog_trm_gin_full_gsn_n50_k3_p0.2_T3_ns3_s${s}.pt echo "===== seed=$s full gsn =====" python3 diag/train_color.py --mode train --grad_mode full --pe gsn --p 0.2 --epochs 150 --seed "$s" \ || echo "!! train gsn s$s failed" python3 diag/train_color.py --mode le --ckpt "$ck" || echo "!! le gsn s$s failed" - python3 diag/ptrm_color.py --ckpt "$ck" --K 16 --n_graphs 150 --sigmas 0.2 || echo "!! ptrm gsn s$s failed" done echo "===== AGGREGATE (all pe: none/rwse/gsn) =====" python3 diag/aggregate.py |