diff --git a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp index efa0ffee4..228f23539 100644 Binary files a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp and b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp differ diff --git a/egs/librispeech/ASR/conformer_ctc2/decode.py b/egs/librispeech/ASR/conformer_ctc2/decode.py index 478f60625..fb5b0295b 100755 --- a/egs/librispeech/ASR/conformer_ctc2/decode.py +++ b/egs/librispeech/ASR/conformer_ctc2/decode.py @@ -238,8 +238,7 @@ def greedy_search( """ ctc_probs = probs _, max_index = ctc_probs.max(2) # (B, maxlen) - print(mask.size()) - max_index = max_index.masked_fill_(mask, 0) # (B, maxlen) + max_index = max_index.masked_fill_(mask[:,:max_index.size(1)], 0) # (B, maxlen) ret_hyps = [] for hyp in max_index: