Package full RRoG GNN project

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)"