diff --git a/egs/aishell/ASR/transformer_ctc/.decode.py.swp b/egs/aishell/ASR/transformer_ctc/.decode.py.swp index 276f8a4b6..e82de0b0c 100644 Binary files a/egs/aishell/ASR/transformer_ctc/.decode.py.swp and b/egs/aishell/ASR/transformer_ctc/.decode.py.swp differ diff --git a/egs/aishell/ASR/transformer_ctc/decode.py b/egs/aishell/ASR/transformer_ctc/decode.py index 46873fe11..cd9069766 100755 --- a/egs/aishell/ASR/transformer_ctc/decode.py +++ b/egs/aishell/ASR/transformer_ctc/decode.py @@ -548,6 +548,12 @@ def main(): num_decoder_layers=params.num_decoder_layers, ) + f = open(f"{params.lang_dir}/tokens.txt", "r").readlines() + token_dict = {} + for line in f: + line = line.strip().split() + token_dict[line[1]] = line[0] + if params.avg == 1: load_checkpoint(f"{params.exp_dir}/epoch-{params.epoch}.pt", model) else: