diff --git a/examples/aishell/run_infer.sh b/examples/aishell/run_infer.sh index c38325d17..6683b770c 100644 --- a/examples/aishell/run_infer.sh +++ b/examples/aishell/run_infer.sh @@ -30,7 +30,7 @@ python -u infer.py \ --infer_manifest='data/aishell/manifest.test' \ --mean_std_path='data/aishell/mean_std.npz' \ --vocab_path='data/aishell/vocab.txt' \ ---model_path='checkpoints/aishell/srep_final' \ +--model_path='checkpoints/aishell/step_final' \ --lang_model_path='models/lm/zh_giga.no_cna_cmn.prune01244.klm' \ --decoding_method='ctc_beam_search' \ --error_rate_type='cer' \ diff --git a/infer.py b/infer.py index e43aa5266..9e6a8c3b2 100644 --- a/infer.py +++ b/infer.py @@ -46,7 +46,7 @@ add_arg('lang_model_path', str, 'models/lm/common_crawl_00.prune01111.trie.klm', "Filepath for language model.") add_arg('model_path', str, - './checkpoints/libri/params.latest.tar.gz', + './checkpoints/libri/step_final', "If None, the training starts from scratch, " "otherwise, it resumes from the pre-trained model.") add_arg('decoding_method', str, diff --git a/test.py b/test.py index 12df55219..ec9fdfda6 100644 --- a/test.py +++ b/test.py @@ -39,7 +39,7 @@ add_arg('vocab_path', str, 'data/librispeech/vocab.txt', "Filepath of vocabulary.") add_arg('model_path', str, - './checkpoints/libri/params.latest.tar.gz', + './checkpoints/libri/step_final', "If None, the training starts from scratch, " "otherwise, it resumes from the pre-trained model.") add_arg('lang_model_path', str,