diff --git a/hydra_flash/conf/config.yaml b/hydra_flash/conf/config.yaml index d1f5fa37e9273ec6defc933db98584548b0d5dd6..815a2e7cc98ff5afb792fe53264431e78f078ce3 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 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/hydra_flash/conf/server/local.yaml b/hydra_flash/conf/server/local.yaml new file mode 100644 index 0000000000000000000000000000000000000000..c6439317590b7cf5a348f4588f9c0e4bb1539cf3 --- /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 0000000000000000000000000000000000000000..66f02e78c96f11c9e24571ad1be93c04017587fe --- /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"