blob: 82c522a924b4509abccadf63c5907fe00dc59d39 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
|
#!/usr/bin/env bash
# RRoG/TRM-on-GNN: graph 3-coloring, deterministic T=1 lower bound vs T=3 extra compute.
set -uo pipefail
cd /home/yurenh2/rrog
export PYTHONPATH=/home/yurenh2/rrog
echo "host=$(hostname) gpu=${CUDA_VISIBLE_DEVICES:-?} start=$(date -Is)"
for T in 1 3; do
echo "===== train T=$T ====="
python3 diag/train_color.py --mode train --grad_mode full --T "$T" --p 0.2 --epochs 150 --seed 0 \
|| echo "!! train T=$T failed"
done
echo "done=$(date -Is)"
|
