diff --git a/reference_algorithms/target_setting_algorithms/librispeech_deepspeech/librispeech_deepspeech_no_resnet/tuning_search_space.json b/reference_algorithms/target_setting_algorithms/librispeech_deepspeech/librispeech_deepspeech_no_resnet/tuning_search_space.json new file mode 100644 index 000000000..b31b711f7 --- /dev/null +++ b/reference_algorithms/target_setting_algorithms/librispeech_deepspeech/librispeech_deepspeech_no_resnet/tuning_search_space.json @@ -0,0 +1,27 @@ +{ + "learning_rate": { + "feasible_points": [ + 0.0035278622506232458 + ] + }, + "beta1": { + "feasible_points": [ + 0.8192305396005781 + ] + }, + "beta2": { + "feasible_points": [ + 0.495850879212151 + ] + }, + "warmup_steps": { + "feasible_points": [ + 6000 + ] + }, + "weight_decay": { + "feasible_points": [ + 0.04339748256184769 + ] + } +} diff --git a/reference_algorithms/target_setting_algorithms/librispeech_deepspeech/librispeech_deepspeech_norm_and_spec_aug/tuning_search_space.json b/reference_algorithms/target_setting_algorithms/librispeech_deepspeech/librispeech_deepspeech_norm_and_spec_aug/tuning_search_space.json new file mode 100644 index 000000000..e20a2dae1 --- /dev/null +++ b/reference_algorithms/target_setting_algorithms/librispeech_deepspeech/librispeech_deepspeech_norm_and_spec_aug/tuning_search_space.json @@ -0,0 +1,27 @@ +{ + "learning_rate": { + "feasible_points": [ + 0.001308209823469072 + ] + }, + "beta1": { + "feasible_points": [ + 0.9731333693827139 + ] + }, + "beta2": { + "feasible_points": [ + 0.9981232922116359 + ] + }, + "warmup_steps": { + "feasible_points": [ + 6000 + ] + }, + "weight_decay": { + "feasible_points": [ + 0.16375311233774334 + ] + } +} diff --git a/scoring/workload_meta_data.json b/scoring/workload_meta_data.json index 1c3f63a77..3a6267ca9 100644 --- a/scoring/workload_meta_data.json +++ b/scoring/workload_meta_data.json @@ -1,10 +1,34 @@ { - "imagenet_resnet": { "max_steps": 186666, "dataset": "imagenet" }, - "imagenet_vit": { "max_steps": 186666, "dataset": "imagenet" }, - "fastmri": { "max_steps": 36189, "dataset": "fastmri" }, - "ogbg": { "max_steps": 80000, "dataset": "ogbg" }, - "wmt": { "max_steps": 133333, "dataset": "wmt" }, - "librispeech_deepspeech": { "max_steps": 48000, "dataset": "librispeech" }, - "criteo1tb": { "max_steps": 10666, "dataset": "criteo1tb" }, - "librispeech_conformer": { "max_steps": 80000, "dataset": "librispeech" } - } \ No newline at end of file + "imagenet_resnet": { + "max_steps": 186666, + "dataset": "imagenet" + }, + "imagenet_vit": { + "max_steps": 186666, + "dataset": "imagenet" + }, + "fastmri": { + "max_steps": 36189, + "dataset": "fastmri" + }, + "ogbg": { + "max_steps": 80000, + "dataset": "ogbg" + }, + "wmt": { + "max_steps": 133333, + "dataset": "wmt" + }, + "librispeech_deepspeech": { + "max_steps": 48000, + "dataset": "librispeech" + }, + "criteo1tb": { + "max_steps": 10666, + "dataset": "criteo1tb" + }, + "librispeech_conformer": { + "max_steps": 80000, + "dataset": "librispeech" + } +} \ No newline at end of file