Package full RRoG GNN project

1 files changed, 5 insertions, 8 deletions

diff --git a/diag/run_color.sh b/diag/run_color.sh
index ad3c406..82c522a 100644
--- a/diag/run_color.sh
+++ b/diag/run_color.sh
@@ -1,15 +1,12 @@
 #!/usr/bin/env bash
-# Step-2 (TRM regime, large output): graph 3-coloring, TRM full vs HRM 1-step + LE diagnostic.
+# RRoG/TRM-on-GNN: graph 3-coloring, deterministic T=1 lower bound vs T=3 extra compute.
 set -uo pipefail
 cd /home/yurenh2/rrog
 export PYTHONPATH=/home/yurenh2/rrog
 echo "host=$(hostname) gpu=${CUDA_VISIBLE_DEVICES:-?} start=$(date -Is)"
-for gm in full 1step; do
-  echo "===== train $gm ====="
-  python3 diag/train_color.py --mode train --grad_mode "$gm" --p 0.2 --epochs 150 --seed 0 \
-    || echo "!! train $gm failed"
+for T in 1 3; do
+  echo "===== train T=$T ====="
+  python3 diag/train_color.py --mode train --grad_mode full --T "$T" --p 0.2 --epochs 150 --seed 0 \
+    || echo "!! train T=$T failed"
 done
-echo "===== LE diagnostic (lambda1: solved vs unsolved) ====="
-python3 diag/train_color.py --mode le --ckpt runs/ckpt_color_full_n50_k3_p0.2_T3_ns3_s0.pt  || echo "!! le full failed"
-python3 diag/train_color.py --mode le --ckpt runs/ckpt_color_1step_n50_k3_p0.2_T3_ns3_s0.pt || echo "!! le 1step failed"
 echo "done=$(date -Is)"