blob: e887a5ec5c5063e470c8da881cad5ca7e1d087fc (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
#!/bin/bash
# Diagnose all 10 HRM checkpoints with joint Lyapunov (512 samples each)
set -e
cd /home/yurenh2/rrm/research/flossing
source /home/yurenh2/miniconda3/etc/profile.d/conda.sh
conda activate rrm
export CUDA_VISIBLE_DEVICES=0
CKPT_ROOT="/home/yurenh2/rrm/hrm/checkpoints/Sudoku-extreme-1k-aug-1000 ACT-torch/HierarchicalReasoningModel_ACTV1 righteous-python"
for STEP in step_2604 step_5208 step_7812 step_10416 step_13020 step_15624 step_18228 step_20832 step_23436 step_26040; do
OUT="diag_hrm_${STEP}_512.npz"
LOG="diag_hrm_${STEP}.log"
if [ -f "$OUT" ]; then
echo "skip $STEP (exists)"
continue
fi
echo "[$(date '+%H:%M:%S')] Starting HRM diagnostic $STEP"
python diagnose_hrm_joint.py \
--ckpt-root "$CKPT_ROOT" --ckpt-name "$STEP" \
--n-samples 512 --batch-size 32 --k-lyap 8 --t-ons 1 --seed 0 \
--out "$OUT" > "$LOG" 2>&1
echo "[$(date '+%H:%M:%S')] $STEP done"
done
echo "[$(date '+%H:%M:%S')] All HRM diagnostics complete"
|
