diff --git a/PyTorch/Segmentation/nnUNet/submit_benchmark_sbatch.sh b/PyTorch/Segmentation/nnUNet/submit_benchmark_sbatch.sh index cafc395ed16b5b2f3a45b6f6b704b4dd15116f46..2e5f89a951c9d423a43e84f820fe0a6168fbe35e 100644 --- a/PyTorch/Segmentation/nnUNet/submit_benchmark_sbatch.sh +++ b/PyTorch/Segmentation/nnUNet/submit_benchmark_sbatch.sh @@ -6,7 +6,7 @@ WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/PyTorch/Segmentation/nnUNet dim=2 for nodes in {1..1}; do for gpus in {1,8}; do - for batch_size in 128; do + for batch_size in 256; do for iteration in {1..1}; do echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, iteration ${iteration}