From fd3a5a06414783606db9b0a6a1b0579f050eaeb6 Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuagu37@gmail.com>
Date: Tue, 18 Oct 2022 13:42:49 +0200
Subject: [PATCH] Update benchmark_nnunet_pytorch_berzelius.sh

---
 benchmark_nnunet_pytorch_berzelius.sh | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/benchmark_nnunet_pytorch_berzelius.sh b/benchmark_nnunet_pytorch_berzelius.sh
index 3f0e00a..4551eb9 100644
--- a/benchmark_nnunet_pytorch_berzelius.sh
+++ b/benchmark_nnunet_pytorch_berzelius.sh
@@ -7,6 +7,9 @@
 #SBATCH --reservation=your-reservation
 #SBATCH -o benchmark.out
 
+#######SBATCH --ntasks-per-node=8
+# Add the above line for multi-node case. Otherwise it will hang.
+
 dim=2
 nodes=1
 
-- 
GitLab