recog with bfloat16

rwth-i6 · Jan 10, 2025 · 872ac06 · 872ac06
1 parent 738d2ec
commit 872ac06
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
@@ -99,8 +99,8 @@ def py():
                     "beam_size": 1024,
                     "beam_size_token": 128,
                     "beam_threshold": 14,
-                    "recog_version": 6,
                     "batch_size": 5_000 * ctc_model.definition.batch_size_factor,
+                    "torch_amp": {"dtype": "bfloat16"},
                 },
                 search_rqmt={"cpu": 4, "mem": 30, "time": 24, "gpu_mem": 24},
             )