mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-09 00:54:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

bug fixes

Browse Source
This commit is contained in:
jinzr 2023-08-14 01:30:13 +08:00
parent 6c088dfa48
commit 5c217dc295
2 changed files with 4 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
egs/aishell4/ASR/zipformer/onnx_decode.py
View File

@ -259,18 +259,13 @@ def main():
c.supervisions[0].text = text_normalize(text) c.supervisions[0].text = text_normalize(text)
return c return c
dev_cuts = aishell4.valid_cuts()
dev_cuts = dev_cuts.map(text_normalize_for_cut)
dev_cuts = dev_cuts.filter(remove_short_utt)
dev_dl = aishell4.valid_dataloaders(dev_cuts)
test_cuts = aishell4.test_cuts() test_cuts = aishell4.test_cuts()
test_cuts = test_cuts.map(text_normalize_for_cut) test_cuts = test_cuts.map(text_normalize_for_cut)
test_cuts = test_cuts.filter(remove_short_utt) test_cuts = test_cuts.filter(remove_short_utt)
test_dl = aishell4.test_dataloaders(test_cuts) test_dl = aishell4.test_dataloaders(test_cuts)
test_sets = ["dev", "test"] test_sets = ["test"]
test_dls = [dev_dl, test_dl] test_dls = [test_dl]
for test_set, test_dl in zip(test_sets, test_dls): for test_set, test_dl in zip(test_sets, test_dls):
start_time = time.time() start_time = time.time()

8
egs/aishell4/ASR/zipformer/streaming_decode.py
View File

@ -866,14 +866,10 @@ def main():
c.supervisions[0].text = text_normalize(text) c.supervisions[0].text = text_normalize(text)
return c return c
dev_cuts = aishell4.valid_cuts()
test_cuts = aishell4.test_cuts() test_cuts = aishell4.test_cuts()
dev_cuts = dev_cuts.map(text_normalize_for_cut)
test_cuts = test_cuts.map(text_normalize_for_cut) test_cuts = test_cuts.map(text_normalize_for_cut)
test_sets = ["test"]
test_sets = ["dev", "test"] test_cuts = [test_cuts]
test_cuts = [dev_cuts, test_cuts]
for test_set, test_cut in zip(test_sets, test_cuts): for test_set, test_cut in zip(test_sets, test_cuts):
results_dict = decode_dataset( results_dict = decode_dataset(