Update 4 files

- /PyTorch/Segmentation/nnUNet/submit_benchmark_sbatch.sh
- /PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh
- /PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh
- /PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh

PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh → PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh

@@ -22,7 +22,4 @@ cat <<EOT >  $SBATCH_DIR
 rm -f $WORK_DIR/results/$LOG_DIR
 apptainer exec --nv -B ${WORK_DIR}/data:/data -B ${WORK_DIR}/results:/results --pwd /workspace/nnunet_pyt $CONTAINER_DIR python scripts/benchmark.py --mode ${5} --gpus ${3} --dim ${1} --batch_size ${4} --amp --logname='$LOG_DIR'
 
-
-EOT
-
-sbatch $SBATCH_DIR
\ No newline at end of file
+EOT
\ No newline at end of file

PyTorch/Segmentation/nnUNet/submit_benchmark_sbatch.sh → PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh

@@ -13,7 +13,9 @@ for nodes in {1..1}; do
                 echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, benchmark_mode ${benchmark_mode}
 
                 # For single node
-                bash $WORK_DIR/run_benchmark_single_node.sh ${dim} ${nodes} ${gpus} ${batch_size} ${benchmark_mode}
+                bash $WORK_DIR/generate_benchmark_jobs.sh ${dim} ${nodes} ${gpus} ${batch_size} ${benchmark_mode}
+                SBATCH_DIR=$WORK_DIR/sbatch_scripts/benchmark_${benchmark_mode}_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}.sbatch
+                sbatch $SBATCH_DIR
                 sleep 1 
                 
             done