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 index 0d6b75ec1..e3e5d7558 100644 --- 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)