From d12722525fc010a3910b5152c72654a2ade5eac4 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Wed, 17 Jun 2026 11:19:27 -0500 Subject: Initial import --- diag/run_le.sh | 12 ++++++++++++ 1 file changed, 12 insertions(+) create mode 100644 diag/run_le.sh (limited to 'diag/run_le.sh') diff --git a/diag/run_le.sh b/diag/run_le.sh new file mode 100644 index 0000000..8fc31ea --- /dev/null +++ b/diag/run_le.sh @@ -0,0 +1,12 @@ +#!/usr/bin/env bash +# Step-2(iii): TRM-ish GIN full-recursion vs 1-step-gradient, then LE diagnostic on each. +set -uo pipefail +cd /home/yurenh2/rrog +export PYTHONPATH=/home/yurenh2/rrog +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" +echo "done=$(date -Is)" -- cgit v1.2.3