Add config for sharded kyiv

wearepal · Oct 3, 2023 · 077a1cf · 077a1cf
1 parent 461c858
commit 077a1cf
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/external_confs/hydra/launcher/slurm/kyiv_shard.yaml b/external_confs/hydra/launcher/slurm/kyiv_shard.yaml
@@ -0,0 +1,12 @@
+---
+# usage: hydra/launcher=slurm/kyiv_shard
+
+defaults:
+  - submitit_slurm
+
+partition: kyiv
+cpus_per_task: 12  # on kyiv, we have 24 CPUs per GPU
+timeout_min: 99999  # 99999 minutes = a few months
+
+additional_parameters:
+  gres: "shard:5"  # 5 shards à 8GB