fix slurm search shell scripts (#90)

open-mmlab · Feb 17, 2022 · 3b6423d · 3b6423d
1 parent 5ec2327
commit 3b6423d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/mmcls/slurm_search_mmcls.sh b/tools/mmcls/slurm_search_mmcls.sh
@@ -11,7 +11,7 @@ GPUS=${GPUS:-4}
 GPUS_PER_NODE=${GPUS_PER_NODE:-8}
 CPUS_PER_TASK=${CPUS_PER_TASK:-12}
 SRUN_ARGS=${SRUN_ARGS:-""}
-PY_ARGS=${@:5}
+PY_ARGS=${@:6}
 
 PYTHONPATH="$(dirname $0)/../..":$PYTHONPATH \
 srun -p ${PARTITION} \