diff --git a/benchmark_nnunet_pytorch_berzelius.sh b/benchmark_nnunet_pytorch_berzelius.sh
index 3f0e00ad9c40648ad78ab2093c6d8bc8ba088d96..4551eb90b37be78ac477d4e82513039ebbeae846 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