diff options
Diffstat (limited to 'diag/run_archB.sh')
| -rw-r--r-- | diag/run_archB.sh | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/diag/run_archB.sh b/diag/run_archB.sh index 317638f..ddc6069 100644 --- a/diag/run_archB.sh +++ b/diag/run_archB.sh @@ -1,21 +1,19 @@ #!/usr/bin/env bash -# GPS transformer backbone + feature axis (gin + lappe / all). 5 seeds. (pin GPU at launch.) +# One-shot GPS encoder + feature axis (gin + lappe / all). 5 seeds. (pin GPU at launch.) set -uo pipefail cd /home/yurenh2/rrog export PYTHONPATH=/home/yurenh2/rrog echo "B start gpu=${CUDA_VISIBLE_DEVICES:-?} $(date -Is)" for s in 0 1 2 3 4; do - ck=runs/ckpt_color_gps_full_none_n50_k3_p0.2_T3_ns3_s${s}.pt + ck=runs/ckpt_color_rrog_trm_gps_full_none_n50_k3_p0.2_T3_ns3_s${s}.pt echo "== B s$s conv=gps ==" python3 diag/train_color.py --mode train --conv gps --pe none --p 0.2 --epochs 150 --seed "$s" || echo "!! train gps s$s" python3 diag/train_color.py --mode le --ckpt "$ck" || echo "!! le gps s$s" - python3 diag/ptrm_color.py --ckpt "$ck" --K 16 --n_graphs 150 --sigmas 0.2 || echo "!! ptrm gps s$s" for pe in lappe all; do - ck2=runs/ckpt_color_gin_full_${pe}_n50_k3_p0.2_T3_ns3_s${s}.pt + ck2=runs/ckpt_color_rrog_trm_gin_full_${pe}_n50_k3_p0.2_T3_ns3_s${s}.pt echo "== B s$s gin pe=$pe ==" python3 diag/train_color.py --mode train --conv gin --pe "$pe" --p 0.2 --epochs 150 --seed "$s" || echo "!! train $pe s$s" python3 diag/train_color.py --mode le --ckpt "$ck2" || echo "!! le $pe s$s" - python3 diag/ptrm_color.py --ckpt "$ck2" --K 16 --n_graphs 150 --sigmas 0.2 || echo "!! ptrm $pe s$s" done done echo "doneB=$(date -Is)" |