blob: c436c4e3472016be8193d1aa64be065f8f71ceb9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
|
#!/usr/bin/env bash
# RRoG/TRM-on-GNN on ZINC ring-counting. Deterministic view-only vs recursive compute.
set -uo pipefail
cd /home/yurenh2/rrog
export PYTHONPATH=/home/yurenh2/rrog
echo "host=$(hostname) gpu=${CUDA_VISIBLE_DEVICES:-?} start=$(date -Is)"
run() { echo "===== $* ====="; python3 diag/train_rec.py "$@" --epochs 40 --hidden 64 --bs 256 --seed 0 || echo "!! FAILED $*"; }
run --grad_mode full --T 0 --n_sup 3 --sigma 0 --K 1 --select none
run --grad_mode full --T 1 --n_sup 3 --sigma 0 --K 1 --select none
run --grad_mode full --T 3 --n_sup 3 --sigma 0 --K 1 --select none
run --grad_mode full --T 1 --n_sup 3 --act --halt_max_steps 8 --halt_target binary --sigma 0 --K 1 --select none
echo "done=$(date -Is)"
|
