diff --git a/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh b/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh index d46d924628e7f2b720ef99c90d3d8effb3cf6af9..0e3718d454691f192f6a7f0161f089d206c7f699 100644 --- a/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh +++ b/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh @@ -3,8 +3,9 @@ MODULE_NAME=nnunet_for_pytorch MODULE_VERSION=21.11.0 WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/PyTorch/Segmentation/nnUNet -SBATCH_DIR=$WORK_DIR/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_iteration${5}.sbatch +SBATCH_DIR=$WORK_DIR/sbatch_scripts/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_iteration${5}.sbatch LOG_DIR=benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_amp_iteration${5}.json +mkdir -p $WORK_DIR/sbatch_out $WORK_DIR/sbatch_scripts cat <<EOT > $SBATCH_DIR #!/bin/bash @@ -21,7 +22,7 @@ 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 -mkdir -p $WORK_DIR/sbatch_out + cd $WORK_DIR