reduce bsz

mlcommons · Sep 25, 2023 · c58851d · c58851d
1 parent a03f96f
commit c58851d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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':