oom

rwth-i6 · Dec 16, 2024 · c2baf28 · c2baf28
1 parent 1f7eee0
commit c2baf28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/users/zeyer/experiments/exp2024_04_23_baselines/ctc_recog_ext.py b/users/zeyer/experiments/exp2024_04_23_baselines/ctc_recog_ext.py
@@ -56,7 +56,7 @@ def py():
             task=task,
             model=model,
             recog_def=model_recog,
-            config={"batch_size": 10_000 * ctc_model.definition.batch_size_factor},
+            config={"batch_size": 5_000 * ctc_model.definition.batch_size_factor},
         )
         tk.register_output(f"{prefix}/recog-priorScale{prior_scale}-lmScale{lm_scale}", res.output)