From 8b7ca604c3474cd93b1f1fd0b50e62e98275f0d0 Mon Sep 17 00:00:00 2001 From: jinzr Date: Sat, 9 Mar 2024 14:09:58 +0800 Subject: [PATCH] minor updates --- egs/mdcc/ASR/zipformer/decode.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/egs/mdcc/ASR/zipformer/decode.py b/egs/mdcc/ASR/zipformer/decode.py index a8fd47927d..ce104baf72 100755 --- a/egs/mdcc/ASR/zipformer/decode.py +++ b/egs/mdcc/ASR/zipformer/decode.py @@ -769,7 +769,7 @@ def main(): # we need cut ids to display recognition results. args.return_cuts = True - aishell = MdccAsrDataModule(args) + mdcc = MdccAsrDataModule(args) def remove_short_utt(c: Cut): T = ((c.num_frames - 7) // 2 + 1) // 2 @@ -779,13 +779,13 @@ def remove_short_utt(c: Cut): ) return T > 0 - valid_cuts = aishell.valid_cuts() + valid_cuts = mdcc.valid_cuts() valid_cuts = valid_cuts.filter(remove_short_utt) - valid_dl = aishell.valid_dataloaders(valid_cuts) + valid_dl = mdcc.valid_dataloaders(valid_cuts) - test_cuts = aishell.test_cuts() + test_cuts = mdcc.test_cuts() test_cuts = test_cuts.filter(remove_short_utt) - test_dl = aishell.test_dataloaders(test_cuts) + test_dl = mdcc.test_dataloaders(test_cuts) test_sets = ["valid", "test"] test_dls = [valid_dl, test_dl]