You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
- #!/usr/bin/env bash
-
- set -x
-
- PARTITION=$1
- JOB_NAME=$2
- CONFIG=$3
- CHECKPOINT=$4
- GPUS=${GPUS:-8}
- GPUS_PER_NODE=${GPUS_PER_NODE:-8}
- CPUS_PER_TASK=${CPUS_PER_TASK:-5}
- PY_ARGS=${@:5}
- SRUN_ARGS=${SRUN_ARGS:-""}
-
- PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
- srun -p ${PARTITION} \
- --job-name=${JOB_NAME} \
- --gres=gpu:${GPUS_PER_NODE} \
- --ntasks=${GPUS} \
- --ntasks-per-node=${GPUS_PER_NODE} \
- --cpus-per-task=${CPUS_PER_TASK} \
- --kill-on-bad-exit=1 \
- ${SRUN_ARGS} \
- python -u tools/test.py ${CONFIG} ${CHECKPOINT} --launcher="slurm" ${PY_ARGS}
|