From 6da35058fab4e7ca76636d01152ffd6a01712dd3 Mon Sep 17 00:00:00 2001 From: qmpzzpmq <405691733@qq.com> Date: Mon, 15 Aug 2022 15:11:00 +0000 Subject: [PATCH] data config update --- .../ASR/conformer_ctc/generate_CTC_label.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/egs/aishell/ASR/conformer_ctc/generate_CTC_label.py b/egs/aishell/ASR/conformer_ctc/generate_CTC_label.py index 83f74b580..a41921caf 100755 --- a/egs/aishell/ASR/conformer_ctc/generate_CTC_label.py +++ b/egs/aishell/ASR/conformer_ctc/generate_CTC_label.py @@ -133,6 +133,8 @@ def main(): parser = get_parser() AishellAsrDataModule.add_arguments(parser) args = parser.parse_args() + args.enable_spec_aug = False + args.enable_musan = False args.exp_dir = Path(args.exp_dir) args.lang_dir = Path(args.lang_dir) @@ -183,19 +185,19 @@ def main(): logging.info(f"Number of model parameters: {num_param}") aishell = AishellAsrDataModule(args) - test_cuts = aishell.test_cuts() - test_dl = aishell.test_dataloaders(test_cuts) + train_cuts = aishell.train_cuts() + train_dl = aishell.train_dataloaders(train_cuts) - test_sets = ["test"] - test_dls = [test_dl] + train_sets = ["train"] + train_dls = [train_dl] - for test_set, test_dl in zip(test_sets, test_dls): + for train_set, train_dl in zip(train_sets, train_dls): generate_ctc_label_dataset( - dl=test_dl, + dl=train_dl, params=params, model=model, device=device, - output_path=os.path.join(args.exp_dir, f"ctc-label-{test_set}.lca"), + output_path=os.path.join(args.exp_dir, f"ctc-label-{train_set}.lca"), ) logging.info("Done!")