✨ feat(config): best run

ziyixi · ziyixi · commit cd7af073e2a8 · 2023-05-09T11:34:48.000-04:00
The new result seems to be the best all the time, just update its configuration here. https://wandb.ai/xiziyi/PhaseNet-TF/runs/ymsgie60?workspace=user-xiziyi
diff --git a/configs/experiment/production_default.yaml b/configs/experiment/production_default.yaml
@@ -11,7 +11,7 @@ defaults:
 # all parameters below will be merged with parameters from default configurations set above
 # this allows you to overwrite only specified parameters
 
-tags: ["production", "2node", "wandb"]
+tags: ["production", "mul_nodes", "wandb"]
 
 seed: 3407
 
diff --git a/configs/hydra/best_run.yaml b/configs/hydra/best_run.yaml
@@ -0,0 +1,11 @@
+defaults:
+  - default.yaml
+
+launcher:
+  timeout_min: 239
+  nodes: 4
+  tasks_per_node: 4
+  cpus_per_task: 5
+  mem_gb: 140
+  gres: gpu:a100:4
+  account: cmse
diff --git a/configs/hydra/slurm_2n8g_a100.yaml b/configs/hydra/slurm_2n8g_a100.yaml
@@ -0,0 +1,11 @@
+defaults:
+  - default.yaml
+
+launcher:
+  timeout_min: 239
+  nodes: 2
+  tasks_per_node: 4
+  cpus_per_task: 5
+  mem_gb: 140
+  gres: gpu:a100:4
+  account: cmse
diff --git a/configs/hydra/slurm_4n16g_a100.yaml b/configs/hydra/slurm_4n16g_a100.yaml
@@ -0,0 +1,11 @@
+defaults:
+  - default.yaml
+
+launcher:
+  timeout_min: 239
+  nodes: 4
+  tasks_per_node: 4
+  cpus_per_task: 5
+  mem_gb: 140
+  gres: gpu:a100:4
+  account: cmse
diff --git a/configs/trainer/best_run.yaml b/configs/trainer/best_run.yaml
@@ -0,0 +1,9 @@
+defaults:
+  - default.yaml
+
+accelerator: gpu
+devices: 4
+num_nodes: 4
+strategy: "ddp_find_unused_parameters_false"
+
+max_epochs: 140
diff --git a/configs/trainer/multiple_gpu_4nodes.yaml b/configs/trainer/multiple_gpu_4nodes.yaml
@@ -0,0 +1,9 @@
+defaults:
+  - default.yaml
+
+accelerator: gpu
+devices: 4
+num_nodes: 4
+strategy: "ddp_find_unused_parameters_false"
+
+max_epochs: 140
