mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-01-25 15:59:33 +09:00
parent 66f762e4a1
commit cddeef92c5
2 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp
View File

Binary file not shown.

16
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py
View File

@ -785,6 +785,20 @@ def main():
test_sets = [f"test-clean-{option}", f"test-other-{option}"] test_sets = [f"test-clean-{option}", f"test-other-{option}"]
test_dl = [test_clean_dl, test_other_dl] 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): for test_set, test_dl in zip(test_sets, test_dl):
results_dict = decode_dataset( results_dict = decode_dataset(
dl=test_dl, dl=test_dl,
@ -807,7 +821,7 @@ def main():
test_set_name=test_set, test_set_name=test_set,
results_dict=results_dict, results_dict=results_dict,
) )
'''
''' '''
test_clean_cuts, test_clean_sets = librispeech.test_clean_cuts(option='user') test_clean_cuts, test_clean_sets = librispeech.test_clean_cuts(option='user')
test_other_cuts, test_other_sets = librispeech.test_other_cuts(option='user') test_other_cuts, test_other_sets = librispeech.test_other_cuts(option='user')