diff --git a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp index 49be39ce1..efa0ffee4 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 2d257874b..478f60625 100755 --- a/egs/librispeech/ASR/conformer_ctc2/decode.py +++ b/egs/librispeech/ASR/conformer_ctc2/decode.py @@ -238,6 +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) ret_hyps = []