more

rwth-i6 · Jan 13, 2025 · c1574bb · c1574bb
1 parent 8118743
commit c1574bb
Showing 1 changed file with 12 additions and 0 deletions.
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
@@ -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,