modify configs for variants

mlcommons · Mar 15, 2024 · 1cbb893 · 1cbb893
1 parent 1ca2709
commit 1cbb893
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 13 deletions.
diff --git a/utils/group_2_variants_target_setting_workload_config.json b/utils/group_2_variants_target_setting_workload_config.json
@@ -5,12 +5,7 @@
         "submission_path": "reference_algorithms/target_setting_algorithms/jax_momentum.py",
         "tuning_search_space": "reference_algorithms/target_setting_algorithms/imagenet_resnet_large_bn_init/tuning_search_space.json"
     },
-    "imagenet_vit_map": {
-        "max_steps": 186666,
-        "dataset": "imagenet",
-        "submission_path": "reference_algorithms/target_setting_algorithms/jax_nadamw.py",
-        "tuning_search_space": "reference_algorithms/target_setting_algorithms/imagenet_vit_map/tuning_search_space.json"
-    },
+v
     "fastmri_model_size": {
         "max_steps": 36189,
         "dataset": "fastmri",

diff --git a/utils/group_4_variants_target_setting_workload_config.json b/utils/group_4_variants_target_setting_workload_config.json
@@ -1,8 +1,21 @@
 {
-    "librispeech_deepspeech_norm_and_spec_aug": {
-        "max_steps": 48900,
-        "dataset": "librispeech",
+    "imagenet_resnet_gelu": {
+        "max_steps": 186666,
+        "dataset": "imagenet",
+        "submission_path": "reference_algorithms/target_setting_algorithms/jax_momentum.py",
+        "tuning_search_space": "reference_algorithms/target_setting_algorithms/imagenet_resnet_gelu/tuning_search_space.json"
+    },
+    "imagenet_resnet_large_bn_init": {
+        "max_steps": 186666,
+        "dataset": "imagenet",
+        "submission_path": "reference_algorithms/target_setting_algorithms/jax_momentum.py",
+        "tuning_search_space": "reference_algorithms/target_setting_algorithms/imagenet_resnet_large_bn_init/tuning_search_space.json"
+    },
+    "imagenet_resnet_silu": {
+        "max_steps": 186666,
+        "dataset": "imagenet",
         "submission_path": "reference_algorithms/target_setting_algorithms/jax_nadamw.py",
-        "tuning_search_space": "reference_algorithms/target_setting_algorithms/librispeech_deepspeech_norm_and_spec_aug/tuning_search_space.json"
+        "tuning_search_space": "reference_algorithms/target_setting_algorithms/imagenet_resnet_silu/tuning_search_space.json"
     }
+
 }
diff --git a/utils/group_5_variants_target_setting_workload_config.json b/utils/group_5_variants_target_setting_workload_config.json
@@ -1,8 +1,20 @@
 {
-    "imagenet_resnet_gelu": {
+    "imagenet_vit_glu": {
         "max_steps": 186666,
         "dataset": "imagenet",
-        "submission_path": "reference_algorithms/target_setting_algorithms/jax_momentum.py",
-        "tuning_search_space": "reference_algorithms/target_setting_algorithms/imagenet_resnet_gelu/tuning_search_space.json"
+        "submission_path": "reference_algorithms/target_setting_algorithms/jax_nadamw.py",
+        "tuning_search_space": "reference_algorithms/target_setting_algorithms/imagenet_vit_glu/tuning_search_space.json"
+    },
+    "imagenet_vit_map": {
+        "max_steps": 186666,
+        "dataset": "imagenet",
+        "submission_path": "reference_algorithms/target_setting_algorithms/jax_nadamw.py",
+        "tuning_search_space": "reference_algorithms/target_setting_algorithms/imagenet_vit_map/tuning_search_space.json"
+    },
+    "imagenet_vit_post_ln": {
+        "max_steps": 186666,
+        "dataset": "imagenet",
+        "submission_path": "reference_algorithms/target_setting_algorithms/jax_nadamw.py",
+        "tuning_search_space": "reference_algorithms/target_setting_algorithms/imagenet_vit_post_ln/tuning_search_space.json"
     }
 }