summaryrefslogtreecommitdiff
path: root/diag/run_rec.sh
diff options
context:
space:
mode:
Diffstat (limited to 'diag/run_rec.sh')
-rw-r--r--diag/run_rec.sh13
1 files changed, 6 insertions, 7 deletions
diff --git a/diag/run_rec.sh b/diag/run_rec.sh
index 632498d..c436c4e 100644
--- a/diag/run_rec.sh
+++ b/diag/run_rec.sh
@@ -1,13 +1,12 @@
#!/usr/bin/env bash
-# Step-2: recursive GNN + full PTRM on ZINC ring-counting. Does per-step noise + best-Q@K
-# selection break the 1-WL counting ceiling that input-RNI couldn't? Averaging vs selection.
+# RRoG/TRM-on-GNN on ZINC ring-counting. Deterministic view-only vs recursive compute.
set -uo pipefail
cd /home/yurenh2/rrog
export PYTHONPATH=/home/yurenh2/rrog
echo "host=$(hostname) gpu=${CUDA_VISIBLE_DEVICES:-?} start=$(date -Is)"
-run() { echo "===== $* ====="; python3 diag/train_rec.py "$@" --epochs 200 --seed 0 || echo "!! FAILED $*"; }
-run --sigma 0 --K 1 --select bestq # deterministic recursive baseline (~1-WL ceiling)
-run --sigma 0.1 --K 8 --select none # averaging over rollouts (RNI-style null)
-run --sigma 0.1 --K 8 --select bestq # PTRM-proper: per-step noise + best-Q@K selection
-run --sigma 0.2 --K 16 --select bestq # scaled noise + rollouts
+run() { echo "===== $* ====="; python3 diag/train_rec.py "$@" --epochs 40 --hidden 64 --bs 256 --seed 0 || echo "!! FAILED $*"; }
+run --grad_mode full --T 0 --n_sup 3 --sigma 0 --K 1 --select none
+run --grad_mode full --T 1 --n_sup 3 --sigma 0 --K 1 --select none
+run --grad_mode full --T 3 --n_sup 3 --sigma 0 --K 1 --select none
+run --grad_mode full --T 1 --n_sup 3 --act --halt_max_steps 8 --halt_target binary --sigma 0 --K 1 --select none
echo "done=$(date -Is)"