runs, restart w/ 4 gpu

mwalmsley · Nov 24, 2023 · 1ec59bd · 1ec59bd
1 parent 076be8e
commit 1ec59bd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/only_for_me/narval/train.sh b/only_for_me/narval/train.sh
@@ -1,7 +1,7 @@
 #!/bin/bash
 #SBATCH --time=23:30:0  
-#SBATCH --nodes=1
-#SBATCH --ntasks-per-node=1
+#SBATCH --nodes=4
+#SBATCH --ntasks-per-node=4
 #SBATCH --cpus-per-task=10
 #SBATCH --mem-per-cpu 4G
 #SBATCH --gres=gpu:v100:1
@@ -50,8 +50,8 @@ REPO_DIR=/project/def-bovy/walml/zoobot
 srun $PYTHON $REPO_DIR/only_for_me/narval/train.py \
     --save-dir $REPO_DIR/only_for_me/narval/desi_300px_maxvit_rmlp_base_rw_224_4gpu_debug \
     --batch-size 32 \
-    --gpus 1 \
-    --nodes 1 \
+    --gpus 4 \
+    --nodes 4 \
     --num-workers 5 \
     --weight-decay 0.05 \
     --architecture maxvit_rmlp_base_rw_224 \