blob: 6365cec8a7b5a26daee360ef78a85e2714a48c9b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
#!/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:0}"
mkdir -p runs logs
python3 -m rrog.cli run \
--task ogbg-molhiv --view gin --compute classic \
--epochs 1 --hidden 32 --bs 64 --seed 991 --device "${DEVICE}" \
--max_train_batches 2 --max_eval_batches 2
python3 -m rrog.cli run \
--task ogbg-molhiv --view gin --compute fixed-rrog \
--epochs 1 --hidden 32 --bs 64 --T 1 --n_sup 2 --seed 992 --device "${DEVICE}" \
--max_train_batches 2 --max_eval_batches 2
|
