|
|
@ -198,8 +198,7 @@ def infer():
|
|
|
|
vocabulary=vocab_list,
|
|
|
|
vocabulary=vocab_list,
|
|
|
|
beam_size=args.beam_size,
|
|
|
|
beam_size=args.beam_size,
|
|
|
|
ext_scoring_func=ext_scorer,
|
|
|
|
ext_scoring_func=ext_scorer,
|
|
|
|
blank_id=len(vocab_list),
|
|
|
|
blank_id=len(vocab_list))
|
|
|
|
num_processes=1)
|
|
|
|
|
|
|
|
for i, beam_search_result in enumerate(beam_search_nproc_results):
|
|
|
|
for i, beam_search_result in enumerate(beam_search_nproc_results):
|
|
|
|
target_transcription = ''.join(
|
|
|
|
target_transcription = ''.join(
|
|
|
|
[vocab_list[index] for index in infer_data[i][1]])
|
|
|
|
[vocab_list[index] for index in infer_data[i][1]])
|
|
|
|