diff options
Diffstat (limited to 'diag/run_color.sh')
| -rw-r--r-- | diag/run_color.sh | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/diag/run_color.sh b/diag/run_color.sh new file mode 100644 index 0000000..ad3c406 --- /dev/null +++ b/diag/run_color.sh @@ -0,0 +1,15 @@ +#!/usr/bin/env bash +# Step-2 (TRM regime, large output): graph 3-coloring, TRM full vs HRM 1-step + LE diagnostic. +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" +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)" |