try 4 gpus

mwalmsley · Nov 4, 2023 · fa5ec25 · fa5ec25
1 parent fa78fc3
commit fa5ec25
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/only_for_me/narval/train.sh b/only_for_me/narval/train.sh
@@ -2,9 +2,9 @@
 #SBATCH --mem=80G
 #SBATCH --nodes=1
 #SBATCH --time=0:40:0  
-#SBATCH --tasks-per-node=2
+#SBATCH --tasks-per-node=4
 #SBATCH --cpus-per-task=12
-#SBATCH --gres=gpu:a100:2
+#SBATCH --gres=gpu:a100:4
 
 nvidia-smi