diff --git a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp index b512b2bb0..a5e051578 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 99165c640..df1b578ea 100755 --- a/egs/librispeech/ASR/conformer_ctc2/decode.py +++ b/egs/librispeech/ASR/conformer_ctc2/decode.py @@ -239,6 +239,7 @@ def greedy_search( ctc_probs = probs print(ctc_probs.size()) _, max_index = ctc_probs.max(2) # (B, maxlen) + print(max_index.size()) max_index = max_index.masked_fill_(mask, 0) # (B, maxlen) ret_hyps = []