From 0a578789697881b2c335dca2376502356e82748a Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Wed, 27 Sep 2023 20:46:54 +0000
Subject: [PATCH] Update file build.txt

---
 PyTorch/Segmentation/nnUNet/build.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/PyTorch/Segmentation/nnUNet/build.txt b/PyTorch/Segmentation/nnUNet/build.txt
index 83c84fd..003ef31 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 
-- 
GitLab