blob: dc9f88b1f9316ce5d527e5a6939eb15554b66dec (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
#!/usr/bin/env bash
# RRoG on a GSN-style motif-augmented one-shot encoder. Full recursion, 5 seeds; train + LE.
set -uo pipefail
cd /home/yurenh2/rrog
export PYTHONPATH=/home/yurenh2/rrog
echo "host=$(hostname) gpu=${CUDA_VISIBLE_DEVICES:-?} start=$(date -Is)"
for s in 0 1 2 3 4; do
ck=runs/ckpt_color_rrog_trm_gin_full_gsn_n50_k3_p0.2_T3_ns3_s${s}.pt
echo "===== seed=$s full gsn ====="
python3 diag/train_color.py --mode train --grad_mode full --pe gsn --p 0.2 --epochs 150 --seed "$s" \
|| echo "!! train gsn s$s failed"
python3 diag/train_color.py --mode le --ckpt "$ck" || echo "!! le gsn s$s failed"
done
echo "===== AGGREGATE (all pe: none/rwse/gsn) ====="
python3 diag/aggregate.py
echo "done=$(date -Is)"
|
