diff --git a/tools/scripts/slurm_test_single.sh b/tools/scripts/slurm_test_single.sh new file mode 100644 index 0000000..2dea3ce --- /dev/null +++ b/tools/scripts/slurm_test_single.sh @@ -0,0 +1,19 @@ +#!/usr/bin/env bash + +set -x + +PARTITION=$1 +GPUS=1 +GPUS_PER_NODE=1 +PY_ARGS=${@:2} +JOB_NAME=eval +SRUN_ARGS=${SRUN_ARGS:-""} + +srun -p ${PARTITION} \ + --job-name=${JOB_NAME} \ + --gres=gpu:${GPUS_PER_NODE} \ + --ntasks=${GPUS} \ + --ntasks-per-node=${GPUS_PER_NODE} \ + --kill-on-bad-exit=1 \ + ${SRUN_ARGS} \ + python -u test.py ${PY_ARGS}