diff --git a/egs/aishell/ASR/transformer_ctc/.decode.py.swp b/egs/aishell/ASR/transformer_ctc/.decode.py.swp index 2a9343f7f..0a9b80c79 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 c984d9796..78ffc1b05 100755 --- a/egs/aishell/ASR/transformer_ctc/decode.py +++ b/egs/aishell/ASR/transformer_ctc/decode.py @@ -264,7 +264,9 @@ def decode_one_batch( if params.method == 'greedy-search' or params.method == 'ctc-decoding': batch_size = nnet_output.size(0) for i in range(batch_size): - print(nnet_output[i].topk(1)) + topk_log_probs, topk_indexes = nnet_output[i].topk(1) + print(topk_indexes) + if params.method == "ctc-decoding": best_path = one_best_decoding(