From c1574bbabd4e1c9043931482879f85d81b246346 Mon Sep 17 00:00:00 2001 From: Albert Zeyer Date: Mon, 13 Jan 2025 18:21:28 +0100 Subject: [PATCH] more --- .../exp2024_04_23_baselines/ctc_recog_ext.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) 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 a9e5391b4..b9b1f351b 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 @@ -117,6 +117,18 @@ def py(): "lm_state_lru_initial_cache_size": 1024, # -- default }, ), + ( + "beam16-beamToken16-cache2pow16", + { + "n_best": 16, + "beam_size": 16, + "beam_size_token": 16, + "beam_threshold": 14, + "batch_size": 5_000 * ctc_model.definition.batch_size_factor, + "torch_amp": {"dtype": "bfloat16"}, + "lm_state_lru_initial_cache_size": 2**16, + }, + ), ]: res = recog_model( task=task,