From c54ddb88b532be28ca3096e21de405d90163ecfa Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Mon, 29 Jun 2026 12:04:47 -0500 Subject: Package full RRoG GNN project --- diag/run_le.sh | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'diag/run_le.sh') diff --git a/diag/run_le.sh b/diag/run_le.sh index 8fc31ea..07d6a63 100644 --- a/diag/run_le.sh +++ b/diag/run_le.sh @@ -1,5 +1,5 @@ #!/usr/bin/env bash -# Step-2(iii): TRM-ish GIN full-recursion vs 1-step-gradient, then LE diagnostic on each. +# RRoG/TRM-on-GNN GIN full-recursion vs 1-step-gradient, then LE diagnostic on each. set -uo pipefail cd /home/yurenh2/rrog export PYTHONPATH=/home/yurenh2/rrog @@ -7,6 +7,6 @@ echo "host=$(hostname) gpu=${CUDA_VISIBLE_DEVICES:-?} start=$(date -Is)" python3 diag/train_rec.py --grad_mode full --sigma 0 --K 1 --epochs 200 --seed 0 || echo "!! train full failed" python3 diag/train_rec.py --grad_mode 1step --sigma 0 --K 1 --epochs 200 --seed 0 || echo "!! train 1step failed" echo "===== LE diagnostic (lambda1: success vs failure) =====" -python3 diag/lyap.py --ckpt runs/ckpt_rec_full_sig0.0_K1_bestq_T3_ns3_s0.pt --n_graphs 300 || echo "!! lyap full failed" -python3 diag/lyap.py --ckpt runs/ckpt_rec_1step_sig0.0_K1_bestq_T3_ns3_s0.pt --n_graphs 300 || echo "!! lyap 1step failed" +python3 diag/lyap.py --ckpt runs/ckpt_rec_rrog_full_sig0.0_K1_bestq_T3_ns3_s0.pt --n_graphs 300 || echo "!! lyap full failed" +python3 diag/lyap.py --ckpt runs/ckpt_rec_rrog_1step_sig0.0_K1_bestq_T3_ns3_s0.pt --n_graphs 300 || echo "!! lyap 1step failed" echo "done=$(date -Is)" -- cgit v1.2.3