#!/usr/bin/env bash
# RRoG on PE-augmented one-shot GIN encoder. GIN vs GIN+RWSE on coloring.
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_rrog_trm_gin_full_${pe}_n50_k3_p0.2_T3_ns3_s0.pt \
    || echo "!! le $pe failed"
done
echo "done=$(date -Is)"