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