Skip to content
Snippets Groups Projects
Commit cd6d8032 authored by Xuan Gu's avatar Xuan Gu
Browse files

Update file run_benchmark_single_node.sh

parent 37bd1251
No related branches found
No related tags found
No related merge requests found
......@@ -19,11 +19,6 @@ cat <<EOT > $SBATCH_DIR
#SBATCH --output=$SBATCH_OUT_DIR
##SBATCH --reservation=devel
MODULE_NAME=nnunet_for_pytorch
MODULE_VERSION=21.11.0
WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/PyTorch/Segmentation/nnUNet
CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODULE_NAME}_${MODULE_VERSION}.sif
rm -f $WORK_DIR/results/$LOG_DIR
apptainer exec --nv -B ${WORK_DIR}/data:/data -B ${WORK_DIR}/results:/results --pwd /workspace/nnunet_pyt $CONTAINER_DIR python scripts/benchmark.py --mode ${5} --gpus ${3} --dim ${1} --batch_size ${4} --amp --logname='$LOG_DIR'
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment