diff options
| author | YurenHao0426 <Blackhao0426@gmail.com> | 2026-06-24 10:22:17 -0500 |
|---|---|---|
| committer | YurenHao0426 <Blackhao0426@gmail.com> | 2026-06-24 10:22:17 -0500 |
| commit | 6cb53c3549400f8784982e9cb724b70dbba183b5 (patch) | |
| tree | 24781dfef04e620732ab7669971446ef7be9ccf1 /scripts/run_ogb_act_two_gpu.sh | |
| parent | 8102b84160d83b0b221505d68a915c746a686ef4 (diff) | |
Allow empty ACT GPU queues
Diffstat (limited to 'scripts/run_ogb_act_two_gpu.sh')
| -rwxr-xr-x | scripts/run_ogb_act_two_gpu.sh | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/scripts/run_ogb_act_two_gpu.sh b/scripts/run_ogb_act_two_gpu.sh index 3c31176..f51262b 100755 --- a/scripts/run_ogb_act_two_gpu.sh +++ b/scripts/run_ogb_act_two_gpu.sh @@ -7,8 +7,12 @@ export PYTHONPATH="${ROOT_DIR}:${PYTHONPATH:-}" GPU0="${GPU0:-cuda:0}" GPU1="${GPU1:-cuda:1}" -TASKS_GPU0="${TASKS_GPU0:-ogbg-molhiv ogbg-molbbbp ogbg-molsider ogbg-molbace}" -TASKS_GPU1="${TASKS_GPU1:-ogbg-molesol ogbg-mollipo ogbg-moltox21 ogbg-molclintox}" +if [[ -z "${TASKS_GPU0+x}" ]]; then + TASKS_GPU0="ogbg-molhiv ogbg-molbbbp ogbg-molsider ogbg-molbace" +fi +if [[ -z "${TASKS_GPU1+x}" ]]; then + TASKS_GPU1="ogbg-molesol ogbg-mollipo ogbg-moltox21 ogbg-molclintox" +fi EPOCHS="${EPOCHS:-100}" SEEDS="${SEEDS:-${SEED:-0}}" HALT_MAX="${HALT_MAX:-8}" |