Update 2 files

- /PyTorch/Segmentation/nnUNet/benchmark.txt
- /PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh

PyTorch/Segmentation/nnUNet/benchmark.txt

@@ -32,4 +32,6 @@ AttributeError: can't set attribute
 
 Solution: pip install torchmetrics==0.6.0, works.
 
+# Muiti-node is not supported in 21.11.0 yet but only in the most recent code on GitHub.
+
 # 2023-09-28: Not working on the devel reservation
\ No newline at end of file

PyTorch/Segmentation/nnUNet/run_benchmark_single_node.sh

@@ -25,7 +25,7 @@ WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/PyTorch/Segmentation/nnUNet
 CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODULE_NAME}_${MODULE_VERSION}.sif
 
 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 train --gpus ${3} --dim ${1} --batch_size ${4} --nodes ${2} --amp --logname='$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 train --gpus ${3} --dim ${1} --batch_size ${4} --amp --logname='$LOG_DIR'
 
 
 EOT