from local

This commit is contained in:
dohe0342 2023-02-25 15:41:29 +09:00
parent c4c2a024ae
commit 9eb75a045f
2 changed files with 1 additions and 1 deletions

View File

@ -236,7 +236,7 @@ def greedy_search(
Returns: Returns:
best path result best path result
""" """
ctc_probs = probs
_, max_index = ctc_probs.max(2) # (B, maxlen) _, max_index = ctc_probs.max(2) # (B, maxlen)
max_index = max_index.masked_fill_(mask, 0) # (B, maxlen) max_index = max_index.masked_fill_(mask, 0) # (B, maxlen)