diff options
Diffstat (limited to 'diag/run_archA.sh')
| -rw-r--r-- | diag/run_archA.sh | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/diag/run_archA.sh b/diag/run_archA.sh index 5385152..6cc6042 100644 --- a/diag/run_archA.sh +++ b/diag/run_archA.sh @@ -1,16 +1,15 @@ #!/usr/bin/env bash -# Conv axis (pe=none): gcn, sage, gat. 5 seeds, train+LE+PTRM. (pin GPU at launch.) +# One-shot aggregation axis (pe=none): gcn, sage, gat. 5 seeds, train+LE. 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 + ck=runs/ckpt_color_rrog_trm_${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)" |