From ed23515fba1fb24c9c018b88fb6e782b0801b93e Mon Sep 17 00:00:00 2001
From: Victor Joos <victor.joos@uclouvain.be>
Date: Tue, 7 Jun 2022 14:37:03 +0200
Subject: [PATCH] add server-specific config

---
 hydra_flash/conf/config.yaml        |  1 +
 hydra_flash/conf/server/__init__.py |  0
 hydra_flash/conf/server/local.yaml  |  1 +
 hydra_flash/conf/server/mb.yaml     | 19 +++++++++++++++++++
 4 files changed, 21 insertions(+)
 create mode 100644 hydra_flash/conf/server/__init__.py
 create mode 100644 hydra_flash/conf/server/local.yaml
 create mode 100644 hydra_flash/conf/server/mb.yaml

diff --git a/hydra_flash/conf/config.yaml b/hydra_flash/conf/config.yaml
index d1f5fa3..815a2e7 100644
--- a/hydra_flash/conf/config.yaml
+++ b/hydra_flash/conf/config.yaml
@@ -1,5 +1,6 @@
 defaults:
   - _self_
+  - server: local
   - datamodule: hymenoptera
   - model: resnet18
   - datamodule@predict_datamodule: hymenoptera_test
diff --git a/hydra_flash/conf/server/__init__.py b/hydra_flash/conf/server/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/hydra_flash/conf/server/local.yaml b/hydra_flash/conf/server/local.yaml
new file mode 100644
index 0000000..c643931
--- /dev/null
+++ b/hydra_flash/conf/server/local.yaml
@@ -0,0 +1 @@
+name: local
diff --git a/hydra_flash/conf/server/mb.yaml b/hydra_flash/conf/server/mb.yaml
new file mode 100644
index 0000000..66f02e7
--- /dev/null
+++ b/hydra_flash/conf/server/mb.yaml
@@ -0,0 +1,19 @@
+# @package _global_
+defaults:
+  - override /hydra/launcher: submitit_slurm
+
+name: manneback
+gpus: [0]
+hydra:
+  launcher:
+    name: hydra_test
+    timeout_min: 360
+    cpus_per_task: 12
+    tasks_per_node: 1
+    signal_delay_s: 120
+    max_num_timeout: 20
+    mem_gb: 32
+    nodes: 1
+    array_parallelism: 4
+    partition: gpu
+    gres: "gpu:1g.10gb:1"
-- 
GitLab