blob: 835016034d398f24fa6f0f8006b7a2491101a587 (
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
|
#!/usr/bin/env bash
# Roadmap #1: RRoG on PE-augmented backbone. GIN vs GIN+RWSE on coloring:
# does RRoG-noise add headroom on top of static structural encoding, or is it redundant?
set -uo pipefail
cd /home/yurenh2/rrog
export PYTHONPATH=/home/yurenh2/rrog
echo "host=$(hostname) gpu=${CUDA_VISIBLE_DEVICES:-?} start=$(date -Is)"
for pe in none rwse; do
echo "===== train full pe=$pe ====="
python3 diag/train_color.py --mode train --grad_mode full --pe "$pe" --p 0.2 --epochs 150 --seed 0 \
|| echo "!! train $pe failed"
done
echo "===== LE (full, both pe) ====="
for pe in none rwse; do
python3 diag/train_color.py --mode le --ckpt runs/ckpt_color_full_${pe}_n50_k3_p0.2_T3_ns3_s0.pt \
|| echo "!! le $pe failed"
done
echo "===== PTRM noise + lambda-select (both pe) ====="
for pe in none rwse; do
echo "--- pe=$pe ---"
python3 diag/ptrm_color.py --ckpt runs/ckpt_color_full_${pe}_n50_k3_p0.2_T3_ns3_s0.pt \
--K 16 --n_graphs 150 --sigmas 0.1 0.2 0.4 || echo "!! ptrm $pe failed"
done
echo "done=$(date -Is)"
|
