#!/usr/bin/env bash
set -uo pipefail
cd /home/yurenh2/rrog
export PYTHONPATH=/home/yurenh2/rrog
echo "PNA start gpu=${CUDA_VISIBLE_DEVICES:-?} $(date -Is)"
for s in 0 1 2 3 4; do
  ck=runs/ckpt_color_rrog_trm_pna_full_none_n50_k3_p0.2_T3_ns3_s${s}.pt
  echo "== pna s$s =="
  python3 diag/train_color.py --mode train --conv pna --pe none --p 0.2 --epochs 150 --seed "$s" || echo "!! train pna s$s"
  python3 diag/train_color.py --mode le --ckpt "$ck" || echo "!! le pna s$s"
done
echo "donePNA=$(date -Is)"