blob: 8d9851f1793821d38c578778d299623d258df676 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
#!/usr/bin/env bash
set -euo pipefail
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
cd "${ROOT_DIR}"
export PYTHONPATH="${ROOT_DIR}:${PYTHONPATH:-}"
ZINC_DEVICE="${ZINC_DEVICE:-cuda:0}"
OGB_DEVICE="${OGB_DEVICE:-cuda:1}"
OGB_TASK="${OGB_TASK:-ogbg-molhiv}"
ZINC_EPOCHS="${ZINC_EPOCHS:-200}"
OGB_EPOCHS="${OGB_EPOCHS:-100}"
SEED="${SEED:-0}"
mkdir -p runs logs
echo "[launch] ZINC-cycle56 on ${ZINC_DEVICE}"
DEVICE="${ZINC_DEVICE}" EPOCHS="${ZINC_EPOCHS}" SEED="${SEED}" \
./scripts/run_zinc_cycle56_full.sh > "logs/zinc_cycle56_${SEED}.log" 2>&1 &
zinc_pid=$!
echo "[launch] ${OGB_TASK} on ${OGB_DEVICE}"
TASK="${OGB_TASK}" DEVICE="${OGB_DEVICE}" EPOCHS="${OGB_EPOCHS}" SEED="${SEED}" \
./scripts/run_ogb_mol_task_full.sh > "logs/${OGB_TASK}_${SEED}.log" 2>&1 &
ogb_pid=$!
echo "[pids] zinc=${zinc_pid} ogb=${ogb_pid}"
wait "${zinc_pid}"
wait "${ogb_pid}"
echo "[done] collecting summaries"
./scripts/collect_results.sh
|
