increase number of gpus back to 40

janosh · Sep 24, 2023 · 603bcb4 · 603bcb4
1 parent 3c0b77d
commit 603bcb4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/models/mace/train_mace.py b/models/mace/train_mace.py
@@ -52,9 +52,9 @@
     slurm_flags="""
         -q preempt
         -C gpu
-        -G 4
-        -N 1
-        --ntasks=4
+        -G 40
+        -N 10
+        --ntasks=40
         --ntasks-per-node=4
         --cpus-per-task=8
         --time=02:00:00
@@ -64,7 +64,7 @@
         --requeue
         --exclusive
         --open-mode=append""",
-    pre_cmd=". ~/.venv/py311/bin/activate;",
+    pre_cmd=". ~/.venv/py311/bin/activate",
 )