diff options
Diffstat (limited to 'diag/run_le.sh')
| -rw-r--r-- | diag/run_le.sh | 12 |
1 files changed, 12 insertions, 0 deletions
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)" |