mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-02-14 15:13:19 +09:00
parent 51b1b74671
commit 771848696f
2 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/tedlium2/ASR/conformer_ctc3/.decode_multi.py.swp
View File

Binary file not shown.

23
egs/tedlium2/ASR/conformer_ctc3/decode_multi.py
View File

@ -778,6 +778,17 @@ def main() -> None:
group_num=params.group_num, group_num=params.group_num,
) )
# we need cut ids to display recognition results.
args.return_cuts = True
tedlium = TedLiumAsrDataModule(args)
valid_cuts = tedlium.dev_cuts()
valid_dl = tedlium.valid_dataloaders(valid_cuts)
test_sets = ["dev"]
test_dls = [valid_dl]
for epoch in range(params.start, params.end+1): for epoch in range(params.start, params.end+1):
load_checkpoint(f"{params.exp_dir}/epoch-{epoch}.pt", model) load_checkpoint(f"{params.exp_dir}/epoch-{epoch}.pt", model)
@ -786,17 +797,7 @@ def main() -> None:
num_param = sum([p.numel() for p in model.parameters()]) num_param = sum([p.numel() for p in model.parameters()])
logging.info(f"Number of model parameters: {num_param}") logging.info(f"Number of model parameters: {num_param}")
# we need cut ids to display recognition results.
args.return_cuts = True
tedlium = TedLiumAsrDataModule(args)
valid_cuts = tedlium.dev_cuts()
valid_dl = tedlium.valid_dataloaders(valid_cuts)
test_sets = ["dev"]
test_dls = [dev_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(
dl=test_dl, dl=test_dl,