mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-14 12:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

data config update

Browse Source
This commit is contained in:
qmpzzpmq 2022-08-15 15:11:00 +00:00
parent 208ac69519
commit 6da35058fa
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
egs/aishell/ASR/conformer_ctc/generate_CTC_label.py
View File

@ -133,6 +133,8 @@ def main():
parser = get_parser() parser = get_parser()
AishellAsrDataModule.add_arguments(parser) AishellAsrDataModule.add_arguments(parser)
args = parser.parse_args() args = parser.parse_args()
args.enable_spec_aug = False
args.enable_musan = False
args.exp_dir = Path(args.exp_dir) args.exp_dir = Path(args.exp_dir)
args.lang_dir = Path(args.lang_dir) args.lang_dir = Path(args.lang_dir)
@ -183,19 +185,19 @@ def main():
logging.info(f"Number of model parameters: {num_param}") logging.info(f"Number of model parameters: {num_param}")
aishell = AishellAsrDataModule(args) aishell = AishellAsrDataModule(args)
test_cuts = aishell.test_cuts() train_cuts = aishell.train_cuts()
test_dl = aishell.test_dataloaders(test_cuts) train_dl = aishell.train_dataloaders(train_cuts)
test_sets = ["test"] train_sets = ["train"]
test_dls = [test_dl] 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( generate_ctc_label_dataset(
dl=test_dl, dl=train_dl,
params=params, params=params,
model=model, model=model,
device=device, 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!") logging.info("Done!")