diff --git a/reference_algorithms/target_setting_algorithms/get_batch_size.py b/reference_algorithms/target_setting_algorithms/get_batch_size.py index 2351e8185..9f76d86d1 100644 --- a/reference_algorithms/target_setting_algorithms/get_batch_size.py +++ b/reference_algorithms/target_setting_algorithms/get_batch_size.py @@ -12,7 +12,7 @@ def get_batch_size(workload_name): elif workload_name == 'imagenet_vit': return 1024 elif workload_name == 'librispeech_conformer': - return 256 + return 32 elif workload_name == 'librispeech_deepspeech': return 256 elif workload_name == 'ogbg':