bug fixes

This commit is contained in:
JinZr 2023-08-14 01:25:32 +08:00
parent 53b7aead10
commit 4b8aefa22f

View File

@ -788,18 +788,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):
results_dict = decode_dataset( results_dict = decode_dataset(