diff --git a/egs/aishell/ASR/transformer_ctc/.decode.py.swp b/egs/aishell/ASR/transformer_ctc/.decode.py.swp index 50ca5f0af..c807eebd3 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 578050232..42496879d 100755 --- a/egs/aishell/ASR/transformer_ctc/decode.py +++ b/egs/aishell/ASR/transformer_ctc/decode.py @@ -268,7 +268,9 @@ def decode_one_batch( topk_log_probs, topk_indexes = nnet_output[i].topk(1) topk_indexes = topk_indexes.squeeze().unique_consecutive() topk_indexes = topk_indexes[topk_indexes != 0] - print(topk_indexes) + hyp = '' + for idx in topk_indexes: + hyp += token_dict[idx] exit()