From cb198117b42ccea3a292e5e04a261a412d1be81f Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Tue, 30 May 2023 12:46:55 +0000
Subject: [PATCH] Update file benchmark_single_node.sbatch
---
scripts/benchmark_single_node.sbatch | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/scripts/benchmark_single_node.sbatch b/scripts/benchmark_single_node.sbatch
index a4c124b..ee25c40 100644
--- a/scripts/benchmark_single_node.sbatch
+++ b/scripts/benchmark_single_node.sbatch
@@ -1,8 +1,9 @@
#!/bin/bash
+sbatch <<EOT
#SBATCH -A nsc
#SBATCH --nodes=1
-#SBATCH --gres=gpu:8
+#SBATCH --gres=gpu:${3}
#SBATCH --time=0-0:10:00
#SBATCH --reservation=devel
@@ -31,3 +32,4 @@ apptainer exec --nv -B ${PWD}/data:/data -B ${PWD}/results:/results nvidia_nnu-n
#rm -f results/benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_amp_iteration${5}.json
#enroot start --rw --mount ${PWD}/data:/data --mount ${PWD}/results:/results nnunet bash -c "cd /workspace/nnunet_pyt && python scripts/benchmark.py --mode train --gpus ${3} --dim ${1} --batch_size ${4} --nodes ${2} --amp --logname='benchmark_dim${1}_nodes${2}_gpus${3}_batchsize${4}_amp_iteration${5}.json'"
+EOT
\ No newline at end of file
--
GitLab