From 6688a4ca09d73cf1be389a710a1da89aa6728821 Mon Sep 17 00:00:00 2001 From: Xuan Gu <xuan.gu@liu.se> Date: Thu, 28 Sep 2023 10:16:38 +0000 Subject: [PATCH] Update file run_benchmark_single_node.sh --- PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh b/PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh index d46d924..0e3718d 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 -- GitLab