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