From ae42ee21677f2bac44357b8da5d571958fe9bb6e Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 2 Feb 2023 18:52:22 +0900 Subject: [PATCH] from local --- .../ASR/transformer_ctc/.decode.py.swp | Bin 36864 -> 36864 bytes egs/aishell/ASR/transformer_ctc/decode.py | 4 +++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/egs/aishell/ASR/transformer_ctc/.decode.py.swp b/egs/aishell/ASR/transformer_ctc/.decode.py.swp index 2a9343f7f65b47517d357c22943362f3c35e64b0..0a9b80c795b623cf2e2324a59099994338f6cc87 100644 GIT binary patch delta 210 zcmZozz|^pSNi4}A%+puFQqO<^2m}}y8rp9s>n_Jsq@={CU^Giz#N=u?MWA#9?8ioiH((;QGG8HoOCd<08Z+?_5rvd=> C{5gLB delta 128 zcmZozz|^pSNi4}A%+puFQqO<^2m}}yX13l=HeRq%%u0Y!eRH6|W`Ew7Tnr2gxgoOZ zlLZszbAot{K-{o-XToBx$(Hf1nhHpupeQr1L?bURwIn{jw4|W4Bsw!zuOz=9Tfm&vM 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(