worked great, now 4x with the split

mwalmsley · Nov 7, 2023 · eb19ada · eb19ada
1 parent 2d8c3bb
commit eb19ada
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/only_for_me/narval/train.py b/only_for_me/narval/train.py
@@ -121,7 +121,7 @@
         resize_after_crop=args.resize_after_crop,
         # hardware parameters
         # gpus=args.gpus,
-        gpus=2,
+        gpus=4,
         nodes=args.nodes,
         mixed_precision=args.mixed_precision,
         wandb_logger=wandb_logger,

diff --git a/only_for_me/narval/train.sh b/only_for_me/narval/train.sh
@@ -1,11 +1,11 @@
 #!/bin/bash
 #SBATCH --time=1:00:0  
 #SBATCH --nodes=1
-#SBATCH --ntasks=2
-#SBATCH --ntasks-per-node=2
+#SBATCH --ntasks=4
+#SBATCH --ntasks-per-node=4
 #SBATCH --cpus-per-task=10
 #SBATCH --mem-per-cpu 4G
-#SBATCH --gres=gpu:a100:2
+#SBATCH --gres=gpu:a100:4
 
 nvidia-smi