blob: ddc60696b699166eaf9311e31197572f9f8c5b8f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
#!/usr/bin/env bash
# One-shot GPS encoder + feature axis (gin + lappe / all). 5 seeds. (pin GPU at launch.)
set -uo pipefail
cd /home/yurenh2/rrog
export PYTHONPATH=/home/yurenh2/rrog
echo "B start gpu=${CUDA_VISIBLE_DEVICES:-?} $(date -Is)"
for s in 0 1 2 3 4; do
ck=runs/ckpt_color_rrog_trm_gps_full_none_n50_k3_p0.2_T3_ns3_s${s}.pt
echo "== B s$s conv=gps =="
python3 diag/train_color.py --mode train --conv gps --pe none --p 0.2 --epochs 150 --seed "$s" || echo "!! train gps s$s"
python3 diag/train_color.py --mode le --ckpt "$ck" || echo "!! le gps s$s"
for pe in lappe all; do
ck2=runs/ckpt_color_rrog_trm_gin_full_${pe}_n50_k3_p0.2_T3_ns3_s${s}.pt
echo "== B s$s gin pe=$pe =="
python3 diag/train_color.py --mode train --conv gin --pe "$pe" --p 0.2 --epochs 150 --seed "$s" || echo "!! train $pe s$s"
python3 diag/train_color.py --mode le --ckpt "$ck2" || echo "!! le $pe s$s"
done
done
echo "doneB=$(date -Is)"
|
