diff options
| -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}" |