diff options
Diffstat (limited to 'diag/run_pe.sh')
| -rw-r--r-- | diag/run_pe.sh | 11 |
1 files changed, 2 insertions, 9 deletions
diff --git a/diag/run_pe.sh b/diag/run_pe.sh index 8350160..1e587ce 100644 --- a/diag/run_pe.sh +++ b/diag/run_pe.sh @@ -1,6 +1,5 @@ #!/usr/bin/env bash -# Roadmap #1: RRoG on PE-augmented backbone. GIN vs GIN+RWSE on coloring: -# does RRoG-noise add headroom on top of static structural encoding, or is it redundant? +# RRoG on PE-augmented one-shot GIN encoder. GIN vs GIN+RWSE on coloring. set -uo pipefail cd /home/yurenh2/rrog export PYTHONPATH=/home/yurenh2/rrog @@ -12,13 +11,7 @@ for pe in none rwse; do done echo "===== LE (full, both pe) =====" for pe in none rwse; do - python3 diag/train_color.py --mode le --ckpt runs/ckpt_color_full_${pe}_n50_k3_p0.2_T3_ns3_s0.pt \ + python3 diag/train_color.py --mode le --ckpt runs/ckpt_color_rrog_trm_gin_full_${pe}_n50_k3_p0.2_T3_ns3_s0.pt \ || echo "!! le $pe failed" done -echo "===== PTRM noise + lambda-select (both pe) =====" -for pe in none rwse; do - echo "--- pe=$pe ---" - python3 diag/ptrm_color.py --ckpt runs/ckpt_color_full_${pe}_n50_k3_p0.2_T3_ns3_s0.pt \ - --K 16 --n_graphs 150 --sigmas 0.1 0.2 0.4 || echo "!! ptrm $pe failed" -done echo "done=$(date -Is)" |