diff --git a/egs/aishell/ASR/transformer_ctc/.decode.py.swp b/egs/aishell/ASR/transformer_ctc/.decode.py.swp index 390afaedb..d9f99841f 100644 Binary files a/egs/aishell/ASR/transformer_ctc/.decode.py.swp and b/egs/aishell/ASR/transformer_ctc/.decode.py.swp differ diff --git a/egs/aishell/ASR/transformer_ctc/decode.py b/egs/aishell/ASR/transformer_ctc/decode.py index 4ce5774d3..f63a1411b 100755 --- a/egs/aishell/ASR/transformer_ctc/decode.py +++ b/egs/aishell/ASR/transformer_ctc/decode.py @@ -265,7 +265,10 @@ def decode_one_batch( batch_size = nnet_output.size(0) for i in range(batch_size): topk_log_probs, topk_indexes = nnet_output[i].topk(1) - print(topk_indexes.squeeze()) + topk_indexes = topk_indexes.squeeze().unique_consecutive() + topk_indexes = topk_indexes[topk_indexes != 0] + print(topk_indexes.size()) + #print(topk_indexes.squeeze()) exit()