mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

update author info

Browse Source
This commit is contained in:
marcoyang 2022-11-02 17:27:31 +08:00
parent 0a46a39e24
commit babcfd4b68
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
egs/librispeech/ASR/lstm_transducer_stateless2/decode.py
View File

@ -1,7 +1,8 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# #
# Copyright 2021-2022 Xiaomi Corporation (Author: Fangjun Kuang, # Copyright 2021-2022 Xiaomi Corporation (Author: Fangjun Kuang,
# Zengwei Yao) # Zengwei Yao,
# Xiaoyu Yang)
# #
# See ../../../../LICENSE for clarification regarding multiple authors # See ../../../../LICENSE for clarification regarding multiple authors
# #
@ -131,7 +132,6 @@ from beam_search import (
greedy_search_batch, greedy_search_batch,
modified_beam_search, modified_beam_search,
modified_beam_search_rnnlm_shallow_fusion, modified_beam_search_rnnlm_shallow_fusion,
) )
from librispeech import LibriSpeech from librispeech import LibriSpeech
from train import add_model_arguments, get_params, get_transducer_model from train import add_model_arguments, get_params, get_transducer_model
@ -386,11 +386,7 @@ def get_parser():
last output linear layer last output linear layer
""", """,
) )
parser.add_argument( parser.add_argument("--ilm-scale", type=float, default=-0.1)
"--ilm-scale",
type=float,
default=-0.1
)
add_model_arguments(parser) add_model_arguments(parser)
return parser return parser
@ -642,9 +638,13 @@ def decode_dataset(
for batch_idx, batch in enumerate(dl): for batch_idx, batch in enumerate(dl):
texts = batch["supervisions"]["text"] texts = batch["supervisions"]["text"]
cut_ids = [cut.id for cut in batch["supervisions"]["cut"]] cut_ids = [cut.id for cut in batch["supervisions"]["cut"]]
total_duration = sum([cut.duration for cut in batch["supervisions"]["cut"]]) total_duration = sum(
[cut.duration for cut in batch["supervisions"]["cut"]]
)
logging.info(f"Decoding {batch_idx}-th batch, batch size is {len(cut_ids)}, total duration is {total_duration}") logging.info(
f"Decoding {batch_idx}-th batch, batch size is {len(cut_ids)}, total duration is {total_duration}"
)
hyps_dict = decode_one_batch( hyps_dict = decode_one_batch(
params=params, params=params,