Update benchmark_nnunet_pytorch_berzelius.sh

fd3a5a06 · Xuan Gu · GitHub · 07130432 · fd3a5a06
Unverified Commit fd3a5a06 authored 2 years ago by Xuan Gu Committed by GitHub 2 years ago
--- 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