diff options
Diffstat (limited to 'diag/run_seeds.sh')
| -rw-r--r-- | diag/run_seeds.sh | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/diag/run_seeds.sh b/diag/run_seeds.sh index 77f22c8..85b67d1 100644 --- a/diag/run_seeds.sh +++ b/diag/run_seeds.sh @@ -1,5 +1,5 @@ #!/usr/bin/env bash -# Harden coloring results with 5 seeds: full-vs-1step solve, LE AUROC, PTRM(sigma=0.2) for none/rwse. +# Harden corrected RRoG coloring results with 5 seeds: full-vs-1step solve + LE AUROC. set -uo pipefail cd /home/yurenh2/rrog export PYTHONPATH=/home/yurenh2/rrog @@ -7,14 +7,11 @@ echo "host=$(hostname) gpu=${CUDA_VISIBLE_DEVICES:-?} start=$(date -Is)" for s in 0 1 2 3 4; do for cfg in "full none" "full rwse" "1step none"; do set -- $cfg; gm=$1; pe=$2 - ck=runs/ckpt_color_${gm}_${pe}_n50_k3_p0.2_T3_ns3_s${s}.pt + ck=runs/ckpt_color_rrog_trm_gin_${gm}_${pe}_n50_k3_p0.2_T3_ns3_s${s}.pt echo "===== seed=$s $gm $pe =====" python3 diag/train_color.py --mode train --grad_mode "$gm" --pe "$pe" --p 0.2 --epochs 150 --seed "$s" \ || echo "!! train $gm $pe s$s failed" python3 diag/train_color.py --mode le --ckpt "$ck" || echo "!! le $gm $pe s$s failed" - if [ "$gm" = "full" ]; then - python3 diag/ptrm_color.py --ckpt "$ck" --K 16 --n_graphs 150 --sigmas 0.2 || echo "!! ptrm $pe s$s failed" - fi done done echo "===== AGGREGATE =====" |