Update 2 files

- /scripts/benchmark_sbatch_submit.sh
- /README.md

README.md

@@ -82,7 +82,6 @@ bash srtips/copy_data_for_benchmark.sh
 - Run the script.   
 You need to modify the script for e.g. the name of your reservation, number of nodes, batch_size, etc. Also, choose either singularity or enroot.
 ```
-mkdir sbatch_out
 bash srtips/benchmark_sbatch_submit.sh
 ```
 

scripts/benchmark_sbatch_submit.sh

 cd /proj/nsc_testing/xuan/Berzelius-nnU-Net-Benchmark
 
-dim=2
+mkdir -p sbatch_out
 
-nodes=1
-for gpus in {1..8}; do
-    for batch_size in 128; do
-        for iteration in {1..100}; do  
+dim=2
+for nodes in {1..1}; do
+    for gpus in {1..8}; do
+        for batch_size in 128; do
+            for iteration in {1..100}; do
 
-            echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, iteration ${iteration}
-	          # For single node
-            sbatch -o sbatch_out/benchmark_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}_iteration${iteration}.out scripts/benchmark_single_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}     
+                echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, iteration ${iteration}
+                # For single node
+                sbatch -o sbatch_out/benchmark_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}_iteration${iteration}.out benchmark_single_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
 
-            sleep 1 # pause to be kind to the scheduler
+                sleep 1 # pause to be kind to the scheduler
 
+            done
         done
     done
 done
 
+
 for nodes in {2..8}; do
     for gpus in {8}; do
         for batch_size in 128; do