from local

2025-12-11 06:55:27 +00:00 · 2023-02-02 18:50:57 +09:00 · 2023-02-02 18:50:57 +09:00 · 3fc8d0f230
commit 3fc8d0f230
parent 7892a0a5a6
3 changed files with 3 additions and 2 deletions
--- a/egs/aishell/ASR/transformer_ctc/.decode.py.swp
+++ b/egs/aishell/ASR/transformer_ctc/.decode.py.swp
--- 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(
--- a/egs/librispeech/ASR/pruned_transducer_stateless2/.beam_search.py.swp
+++ b/egs/librispeech/ASR/pruned_transducer_stateless2/.beam_search.py.swp