Create benchmark_sbatch_submit.sh

scripts/benchmark_sbatch_submit.sh

+cd /proj/nsc/users/xuan/ngc/DeepLearningExamples/PyTorch/Segmentation/nnUNet
+
+dim=2
+nodes=1
+
+for gpus in 8; do
+
+    for batch_size in 256; do
+
+        for iteration in {1..10}; 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 benchmark_single_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
+            
+            # For multi node
+            #sbatch -o sbatch_out/benchmark_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}_iteration${iteration}.out benchmark_multi_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}           
+            
+            sleep 1 # pause to be kind to the scheduler
+
+        done
+
+    done
+
+done