From cddeef92c5941791d415e5efb636a2e04b028336 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Wed, 25 Jan 2023 15:59:33 +0900 Subject: [PATCH] from local --- .../.decode.py.swp | Bin 40960 -> 45056 bytes .../decode.py | 16 +++++++++++++++- 2 files changed, 15 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp index 71b38efcb444dc4b1cc6a26b18ebd12078ea3929..429a25d2b7790aef7282b2dc1920388b9d029fbe 100644 GIT binary patch delta 698 zcmZwFPe>GD7{~Evc4k{wn{igu#UdLL96J7fWQjtPnwm z4D78Gq%ILk&_a!ex~(39=+H_h1*t=Xmlc)-!wCHj+N}pZy#L{Sf4t-Q!1(t-&CS@Q zb6rmB@t9Dy$ic6LzN>F*ij`Ve>2^_TRwX-HRqn4+t-Fmly0>NjwB9*}ZN3hPJS$^? z)6FUp`;whvPjS~oBd7sEUp5qZFFos?v5J3prL6J?o!bSAp zuOTvrSwjZo4xO`TM>R^iNFGm-!4)Y9Wxm1}8GTWL!bp40|MlD9{P_GYcUHWK!a`E^tsN@P?j z5%)eP;seTn7oI|&LSxt0cp)~FVtD57TPZf%PAQ%2d+ngsSA z+N#w-G$M#V6xxa#Y-;QaJ@~*I-teMfB6?0_w}bV{nqMjeLR^vLYj3C6%ElAQ(Uu#l zsvfyHGwr=+PKW-zZ@c4tB~r8g*H!HW-J^3y_uSqKE67M7Li+|4Q7ybaCX(&g?Sf3iJQ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py index 6db0f4f65..51f0f4ccd 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py @@ -785,6 +785,20 @@ def main(): test_sets = [f"test-clean-{option}", f"test-other-{option}"] test_dl = [test_clean_dl, test_other_dl] + for test_set, test_dl in zip(test_sets, test_dl): + results_dict = decode_dataset( + dl=test_dl, + params=params, + model=model, + sp=sp, + word_table=word_table, + decoding_graph=decoding_graph, + ) + results = results_dict['greedy_search'] + for res in results: + print(res[0], ' '.join(res[1])) + + ''' for test_set, test_dl in zip(test_sets, test_dl): results_dict = decode_dataset( dl=test_dl, @@ -807,7 +821,7 @@ def main(): test_set_name=test_set, results_dict=results_dict, ) - + ''' ''' test_clean_cuts, test_clean_sets = librispeech.test_clean_cuts(option='user') test_other_cuts, test_other_sets = librispeech.test_other_cuts(option='user')