diff --git a/PyTorch/Segmentation/nnUNet/build.txt b/PyTorch/Segmentation/nnUNet/build.txt
index 83c84fd32287259851452b104f8515d65786cb0f..003ef31931d996cc6e6db9d620cfd99eb723fb4e 100644
--- a/PyTorch/Segmentation/nnUNet/build.txt
+++ b/PyTorch/Segmentation/nnUNet/build.txt
@@ -28,7 +28,7 @@ apptainer exec --nv --pwd /workspace/nnunet_pyt $CONTAINER_DIR python download.p
 apptainer exec --nv --pwd /workspace/nnunet_pyt $CONTAINER_DIR  python /workspace/nnunet_pyt/preprocess.py --task 01 --dim 2
  
 
-apptainer exec --nv --pwd /workspace/nnunet_pyt $CONTAINER_DIR python scripts/benchmark.py --mode train --gpus 1 --dim 2 --batch_size 128 --amp
+apptainer exec --nv --pwd /workspace/nnunet_pyt $CONTAINER_DIR python scripts/benchmark.py --mode train --gpus 1 --dim 2 --batch_size 256 --amp
 
 
 # Known issue