diff --git a/egs/aishell/ASR/transformer_ctc/.decode.py.swp b/egs/aishell/ASR/transformer_ctc/.decode.py.swp index 334fd3fa6..2a9343f7f 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 8737d62b2..c984d9796 100755 --- a/egs/aishell/ASR/transformer_ctc/decode.py +++ b/egs/aishell/ASR/transformer_ctc/decode.py @@ -262,8 +262,9 @@ def decode_one_batch( ''' if params.method == 'greedy-search' or params.method == 'ctc-decoding': - print(nnet_output.size()) - exit() + batch_size = nnet_output.size(0) + for i in range(batch_size): + print(nnet_output[i].topk(1)) if params.method == "ctc-decoding": best_path = one_best_decoding( diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/.beam_search.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless2/.beam_search.py.swp index 07ac97abc..87d5b10de 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless2/.beam_search.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless2/.beam_search.py.swp differ