diff --git a/egs/aishell/ASR/transformer_ctc/.decode.py.swp b/egs/aishell/ASR/transformer_ctc/.decode.py.swp index c224253ef..390afaedb 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 bc5b7967b..4ce5774d3 100755 --- a/egs/aishell/ASR/transformer_ctc/decode.py +++ b/egs/aishell/ASR/transformer_ctc/decode.py @@ -265,7 +265,7 @@ 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.size()) + print(topk_indexes.squeeze()) exit()