From e42f575050efeeccb736385b43bed84e1129edb0 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Sun, 21 Jun 2026 15:33:22 -0500 Subject: Initial RRoG GNN runner --- rrog/run_zinc_gine.sh | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100755 rrog/run_zinc_gine.sh (limited to 'rrog/run_zinc_gine.sh') diff --git a/rrog/run_zinc_gine.sh b/rrog/run_zinc_gine.sh new file mode 100755 index 0000000..1ca36e1 --- /dev/null +++ b/rrog/run_zinc_gine.sh @@ -0,0 +1,36 @@ +#!/usr/bin/env bash +set -euo pipefail + +ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)" +cd "${ROOT_DIR}" +export PYTHONPATH="${ROOT_DIR}:${PYTHONPATH:-}" + +DEVICE="${DEVICE:-cuda:3}" +EPOCHS="${EPOCHS:-200}" +SEED="${SEED:-0}" + +run_if_missing() { + local compute="$1" + local t="$2" + local ns="$3" + local out="runs/rec_rrog_gine_full_sig0.0_K1_none_T${t}_ns${ns}_trace_s${SEED}.json" + if [[ -f "${out}" ]]; then + echo "[skip] ${out}" + return + fi + echo "[run] zinc-cycle56 view=gine compute=${compute} T=${t} ns=${ns} device=${DEVICE}" + python3 -m rrog.cli run \ + --task zinc-cycle56 \ + --view gine \ + --compute "${compute}" \ + --epochs "${EPOCHS}" \ + --T "${t}" \ + --n_sup "${ns}" \ + --seed "${SEED}" \ + --device "${DEVICE}" +} + +run_if_missing classic 0 1 +run_if_missing fixed-rrog 1 3 + +python3 -m rrog.cli zinc-results --epochs "${EPOCHS}" -- cgit v1.2.3