mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-08 17:42:21 +00:00
update author info
This commit is contained in:
parent
0a46a39e24
commit
babcfd4b68
@ -1,7 +1,8 @@
|
||||
#!/usr/bin/env python3
|
||||
#
|
||||
# Copyright 2021-2022 Xiaomi Corporation (Author: Fangjun Kuang,
|
||||
# Zengwei Yao)
|
||||
# Zengwei Yao,
|
||||
# Xiaoyu Yang)
|
||||
#
|
||||
# See ../../../../LICENSE for clarification regarding multiple authors
|
||||
#
|
||||
@ -131,7 +132,6 @@ from beam_search import (
|
||||
greedy_search_batch,
|
||||
modified_beam_search,
|
||||
modified_beam_search_rnnlm_shallow_fusion,
|
||||
|
||||
)
|
||||
from librispeech import LibriSpeech
|
||||
from train import add_model_arguments, get_params, get_transducer_model
|
||||
@ -386,11 +386,7 @@ def get_parser():
|
||||
last output linear layer
|
||||
""",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--ilm-scale",
|
||||
type=float,
|
||||
default=-0.1
|
||||
)
|
||||
parser.add_argument("--ilm-scale", type=float, default=-0.1)
|
||||
add_model_arguments(parser)
|
||||
|
||||
return parser
|
||||
@ -642,9 +638,13 @@ def decode_dataset(
|
||||
for batch_idx, batch in enumerate(dl):
|
||||
texts = batch["supervisions"]["text"]
|
||||
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(
|
||||
params=params,
|
||||
|
Loading…
x
Reference in New Issue
Block a user