diff --git a/examples/aishell/run_infer.sh b/examples/aishell/run_infer.sh index c38325d17ebbe695936526a51492c1e39f9847d7..6683b770c35717333bc747c31521697989baea56 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 e43aa52668614e35463fa59f660cfe5a0c6964f1..9e6a8c3b29a4b9db2acfbe26c7de1648e31978a8 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 12df55219bc788aee4e445afbd2ccb17e5767bba..ec9fdfda60d2e62f3b8138155e02c3edf607dd94 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,