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

This commit is contained in:
Fangjun Kuang 2022-09-13 10:59:27 +08:00 committed by GitHub
parent 9e24642faf
commit 145c44f710
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -322,7 +322,7 @@ def main():
H = k2.ctc_topo(
max_token=max_token_id,
modified=False,
modified=params.num_classes > 500,
device=device,
)