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

Update streaming_ctc_decode.py

Browse Source
This commit is contained in:
jinzr 2023-12-07 11:12:47 +08:00
parent 7fb3b13066
commit 206f2143e8
1 changed files with 17 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
egs/librispeech/ASR/zipformer/streaming_ctc_decode.py
View File

@ -128,7 +128,7 @@ def get_parser():
parser.add_argument( parser.add_argument(
"--bpe-model", "--bpe-model",
type=str, type=str,
default="data/lang_bpe_2000/bpe.model", default="data/lang_bpe_500/bpe.model",
help="Path to the BPE model", help="Path to the BPE model",
) )
@ -196,6 +196,22 @@ def get_parser():
return parser return parser
def get_decoding_params() -> AttributeDict:
"""Parameters for decoding."""
params = AttributeDict(
{
"feature_dim": 80,
"frame_shift_ms": 10,
"search_beam": 20,
"output_beam": 8,
"min_active_states": 30,
"max_active_states": 10000,
"use_double_scores": True,
}
)
return params
def decode_one_chunk( def decode_one_chunk(
params: AttributeDict, params: AttributeDict,
model: nn.Module, model: nn.Module,
@ -463,21 +479,6 @@ def save_results(
logging.info(s) logging.info(s)
def get_decoding_params() -> AttributeDict:
"""Parameters for decoding."""
params = AttributeDict(
{
"frame_shift_ms": 10,
"search_beam": 20,
"output_beam": 8,
"min_active_states": 30,
"max_active_states": 10000,
"use_double_scores": True,
}
)
return params
@torch.no_grad() @torch.no_grad()
def main(): def main():
parser = get_parser() parser = get_parser()