from local

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

View File

@ -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')