mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-10 22:45:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
AmirHussein96 2025-09-13 10:40:45 -04:00
parent 947ae0a73c
commit 229883e828
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/iwslt22_ta/ST/zipformer/decode.py
View File

@ -72,7 +72,7 @@ from beam_search import (
greedy_search_batch, greedy_search_batch,
modified_beam_search, modified_beam_search,
) )
from train_st import add_model_arguments, get_params, get_transducer_model from train import add_model_arguments, get_params, get_transducer_model
from icefall.checkpoint import ( from icefall.checkpoint import (
average_checkpoints, average_checkpoints,
@ -529,7 +529,7 @@ def decode_dataset(
word_table=word_table, word_table=word_table,
batch=batch, batch=batch,
) )
#breakpoint()
for name, hyps in hyps_dict.items(): for name, hyps in hyps_dict.items():
this_batch = [] this_batch = []
assert len(hyps) == len(texts) assert len(hyps) == len(texts)
@ -540,7 +540,7 @@ def decode_dataset(
this_batch.append((cut_id, ref_words, ref_words_tgt, hyp_words)) this_batch.append((cut_id, ref_words, ref_words_tgt, hyp_words))
results[name].extend(this_batch) results[name].extend(this_batch)
#breakpoint()
num_cuts += len(texts) num_cuts += len(texts)
if batch_idx % log_interval == 0: if batch_idx % log_interval == 0: