Fix using G before assignment in pruned_transducer_stateless/decode.py

This commit is contained in:
Fangjun Kuang 2022-07-25 20:45:52 +08:00
parent b1d0956855
commit 173a3c48c9

View File

@ -999,6 +999,7 @@ def main():
model.device = device model.device = device
model.unk_id = params.unk_id model.unk_id = params.unk_id
G = None
if "fast_beam_search" in params.decoding_method: if "fast_beam_search" in params.decoding_method:
if params.decoding_method == "fast_beam_search_nbest_LG": if params.decoding_method == "fast_beam_search_nbest_LG":
lexicon = Lexicon(params.lang_dir) lexicon = Lexicon(params.lang_dir)