Update 2 files

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

PyTorch/Segmentation/nnUNet/benchmark.txt

@@ -20,7 +20,7 @@ apptainer exec --nv -B ${WORK_DIR}/data:/data -B ${WORK_DIR}/results:/results --
 
 apptainer exec --nv -B ${WORK_DIR}/data:/data -B ${WORK_DIR}/results:/results --pwd /workspace/nnunet_pyt $CONTAINER_DIR python scripts/benchmark.py --mode predict --gpus 1 --dim 2 --batch_size 256 --amp
 
-
+################# Issues #################
 # Known issue https://github.com/NVIDIA/DeepLearningExamples/issues/1113
 ImportError: cannot import name 'get_num_classes' from 'torchmetrics.utilities.data' (/opt/conda/lib/python3.8/site-packages/torchmetrics/utilities/data.py)
 

PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh

@@ -17,7 +17,7 @@ cat <<EOT >  $SBATCH_DIR
 #SBATCH --gpus=${3}
 #SBATCH --time=0-0:10:00
 #SBATCH --output=$SBATCH_OUT_DIR
-##SBATCH --reservation=devel
+#SBATCH --reservation=devel
 
 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'