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

Minor fixes

Browse Source
This commit is contained in:
pkufool 2023-08-28 16:23:24 +08:00
parent 5bf8d3e857
commit b81fb5a07a
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
egs/librispeech/ASR/zipformer/decode.py
View File

@ -216,6 +216,7 @@ def get_parser():
- greedy_search - greedy_search
- beam_search - beam_search
- modified_beam_search - modified_beam_search
- modified_beam_search_LODR
- fast_beam_search - fast_beam_search
- fast_beam_search_nbest - fast_beam_search_nbest
- fast_beam_search_nbest_oracle - fast_beam_search_nbest_oracle
@ -354,7 +355,8 @@ def get_parser():
default=2, default=2,
help=""" help="""
The bonus score of each token for the context biasing words/phrases. The bonus score of each token for the context biasing words/phrases.
Used only when --decoding_method is modified_beam_search. Used only when --decoding-method is modified_beam_search and
modified_beam_search_LODR.
""", """,
) )
@ -364,7 +366,8 @@ def get_parser():
default="", default="",
help=""" help="""
The path of the context biasing lists, one word/phrase each line The path of the context biasing lists, one word/phrase each line
Used only when --decoding_method is modified_beam_search. Used only when --decoding-method is modified_beam_search and
modified_beam_search_LODR.
""", """,
) )
add_model_arguments(parser) add_model_arguments(parser)
@ -616,8 +619,6 @@ def decode_one_batch(
else: else:
if params.has_contexts: if params.has_contexts:
prefix += f"-context-score-{params.context_score}" prefix += f"-context-score-{params.context_score}"
else:
prefix += "-no-context-words"
return {prefix: hyps} return {prefix: hyps}
else: else:
return {f"beam_size_{params.beam_size}": hyps} return {f"beam_size_{params.beam_size}": hyps}
@ -650,7 +651,7 @@ def decode_dataset(
The word symbol table. The word symbol table.
decoding_graph: decoding_graph:
The decoding graph. Can be either a `k2.trivial_graph` or HLG, Used The decoding graph. Can be either a `k2.trivial_graph` or HLG, Used
only when --decoding_method is fast_beam_search, fast_beam_search_nbest, only when --decoding-method is fast_beam_search, fast_beam_search_nbest,
fast_beam_search_nbest_oracle, and fast_beam_search_nbest_LG. fast_beam_search_nbest_oracle, and fast_beam_search_nbest_LG.
Returns: Returns:
Return a dict, whose key may be "greedy_search" if greedy search Return a dict, whose key may be "greedy_search" if greedy search
@ -814,8 +815,6 @@ def main():
): ):
if params.has_contexts: if params.has_contexts:
params.suffix += f"-context-score-{params.context_score}" params.suffix += f"-context-score-{params.context_score}"
else:
params.suffix += "-no-context-words"
else: else:
params.suffix += f"-context-{params.context_size}" params.suffix += f"-context-{params.context_size}"
params.suffix += f"-max-sym-per-frame-{params.max_sym_per_frame}" params.suffix += f"-max-sym-per-frame-{params.max_sym_per_frame}"