Use modified ctc topo when vocab size is > 500 (#568)

k2-fsa · Sep 13, 2022 · 145c44f · 145c44f
1 parent 9e24642
commit 145c44f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/egs/librispeech/ASR/conformer_ctc/pretrained.py b/egs/librispeech/ASR/conformer_ctc/pretrained.py
@@ -322,7 +322,7 @@ def main():
 
         H = k2.ctc_topo(
             max_token=max_token_id,
-            modified=False,
+            modified=params.num_classes > 500,
             device=device,
         )