fastmri tuning search

mlcommons · Feb 10, 2024 · bcb80ca · bcb80ca
1 parent f9892b1
commit bcb80ca
Show file tree

Hide file tree

Showing 3 changed files with 111 additions and 0 deletions.
diff --git a/reference_algorithms/target_setting_algorithms/fastmri_layernorm/tuning_search_space.json b/reference_algorithms/target_setting_algorithms/fastmri_layernorm/tuning_search_space.json
@@ -0,0 +1,37 @@
+{
+    "learning_rate": {
+        "feasible_points": [
+            0.008334676559764446
+        ]
+    },
+    "beta1": {
+        "feasible_points": [
+            0.8294338711079317
+        ]
+    },
+    "beta2": {
+        "feasible_points": [
+            0.8551723332825868
+        ]
+    },
+    "warmup_steps": {
+        "feasible_points": [
+            2714
+        ]
+    },
+    "decay_steps_factor": {
+        "feasible_points": [
+            0.984398
+        ]
+    },
+    "end_factor": {
+        "feasible_points": [
+            0.01
+        ]
+    },
+    "weight_decay": {
+        "feasible_points": [
+            0.01371235755699044
+        ]
+    }
+}
diff --git a/reference_algorithms/target_setting_algorithms/fastmri_model_size/tuning_search_space.json b/reference_algorithms/target_setting_algorithms/fastmri_model_size/tuning_search_space.json
@@ -0,0 +1,37 @@
+{
+    "learning_rate": {
+        "feasible_points": [
+            0.028609
+        ]
+    },
+    "beta1": {
+        "feasible_points": [
+            0.981543
+        ]
+    },
+    "beta2": {
+        "feasible_points": [
+            0.9978504782314613
+        ]
+    },
+    "warmup_steps": {
+        "feasible_points": [
+            1357
+        ]
+    },
+    "decay_steps_factor": {
+        "feasible_points": [
+            0.984398
+        ]
+    },
+    "end_factor": {
+        "feasible_points": [
+            0.01
+        ]
+    },
+    "weight_decay": {
+        "feasible_points": [
+            0.000576
+        ]
+    }
+}
diff --git a/reference_algorithms/target_setting_algorithms/fastmri_tanh/tuning_search_space.json b/reference_algorithms/target_setting_algorithms/fastmri_tanh/tuning_search_space.json
@@ -0,0 +1,37 @@
+{
+    "learning_rate": {
+        "feasible_points": [
+            0.028609
+        ]
+    },
+    "beta1": {
+        "feasible_points": [
+            0.981543
+        ]
+    },
+    "beta2": {
+        "feasible_points": [
+            0.9978504782314613
+        ]
+    },
+    "warmup_steps": {
+        "feasible_points": [
+            1357
+        ]
+    },
+    "decay_steps_factor": {
+        "feasible_points": [
+            0.984398
+        ]
+    },
+    "end_factor": {
+        "feasible_points": [
+            0.01
+        ]
+    },
+    "weight_decay": {
+        "feasible_points": [
+            0.000576
+        ]
+    }
+}