From 19998ea29b7726c9a5868b2ea8baffc62933ae09 Mon Sep 17 00:00:00 2001 From: huangyuxin Date: Fri, 22 Apr 2022 12:59:50 +0000 Subject: [PATCH 1/4] add aishell conformer, test=doc --- paddlespeech/cli/asr/pretrained_models.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/paddlespeech/cli/asr/pretrained_models.py b/paddlespeech/cli/asr/pretrained_models.py index cc52c751b68..380163f50a6 100644 --- a/paddlespeech/cli/asr/pretrained_models.py +++ b/paddlespeech/cli/asr/pretrained_models.py @@ -27,6 +27,16 @@ 'ckpt_path': 'exp/conformer/checkpoints/wenetspeech', }, + "conformer_aishell-zh-16k": { + 'url': + 'https://paddlespeech.bj.bcebos.com/s2t/aishell/asr1/asr1_conformer_aishell_ckpt_0.1.2.model.tar.gz', + 'md5': + 'ca2c97242863ed4d8baa394be715d989', + 'cfg_path': + 'model.yaml', + 'ckpt_path': + 'exp/conformer/checkpoints/avg_30', + } "transformer_librispeech-en-16k": { 'url': 'https://paddlespeech.bj.bcebos.com/s2t/librispeech/asr1/asr1_transformer_librispeech_ckpt_0.1.1.model.tar.gz', From 12d2f6ea95577ddc182b37004ffd74b9366544c9 Mon Sep 17 00:00:00 2001 From: huangyuxin Date: Sun, 24 Apr 2022 04:56:59 +0000 Subject: [PATCH 2/4] fix conformer_aishell of cli, test=doc --- paddlespeech/cli/asr/pretrained_models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddlespeech/cli/asr/pretrained_models.py b/paddlespeech/cli/asr/pretrained_models.py index 380163f50a6..652f0c431cd 100644 --- a/paddlespeech/cli/asr/pretrained_models.py +++ b/paddlespeech/cli/asr/pretrained_models.py @@ -31,12 +31,12 @@ 'url': 'https://paddlespeech.bj.bcebos.com/s2t/aishell/asr1/asr1_conformer_aishell_ckpt_0.1.2.model.tar.gz', 'md5': - 'ca2c97242863ed4d8baa394be715d989', + '3f073eccfa7bb14e0c6867d65fc0dc3a', 'cfg_path': 'model.yaml', 'ckpt_path': 'exp/conformer/checkpoints/avg_30', - } + }, "transformer_librispeech-en-16k": { 'url': 'https://paddlespeech.bj.bcebos.com/s2t/librispeech/asr1/asr1_transformer_librispeech_ckpt_0.1.1.model.tar.gz', From ad4e04fc82d2b7dd719447682186ea9876c34d78 Mon Sep 17 00:00:00 2001 From: huangyuxin Date: Sun, 24 Apr 2022 05:27:16 +0000 Subject: [PATCH 3/4] add conformer_online_aishell, test=doc --- examples/aishell/asr1/conf/chunk_conformer.yaml | 2 +- paddlespeech/cli/asr/pretrained_models.py | 10 ++++++++++ 2 files changed, 11 insertions(+), 1 deletion(-) diff --git a/examples/aishell/asr1/conf/chunk_conformer.yaml b/examples/aishell/asr1/conf/chunk_conformer.yaml index 9f70e4c576d..3cfe9b1b00e 100644 --- a/examples/aishell/asr1/conf/chunk_conformer.yaml +++ b/examples/aishell/asr1/conf/chunk_conformer.yaml @@ -78,7 +78,7 @@ num_encs: 1 ########################################### # Training # ########################################### -n_epoch: 180 +n_epoch: 240 accum_grad: 1 global_grad_clip: 5.0 dist_sampler: True diff --git a/paddlespeech/cli/asr/pretrained_models.py b/paddlespeech/cli/asr/pretrained_models.py index 652f0c431cd..cb4c5e27728 100644 --- a/paddlespeech/cli/asr/pretrained_models.py +++ b/paddlespeech/cli/asr/pretrained_models.py @@ -37,6 +37,16 @@ 'ckpt_path': 'exp/conformer/checkpoints/avg_30', }, + "conformer_online_aishell-zh-16k": { + 'url': + 'https://paddlespeech.bj.bcebos.com/s2t/aishell/asr1/asr1_chunk_conformer_aishell_ckpt_0.2.0.model.tar.gz', + 'md5': + 'b374cfb93537761270b6224fb0bfc26a', + 'cfg_path': + 'model.yaml', + 'ckpt_path': + 'exp/chunk_conformer/checkpoints/avg_30', + }, "transformer_librispeech-en-16k": { 'url': 'https://paddlespeech.bj.bcebos.com/s2t/librispeech/asr1/asr1_transformer_librispeech_ckpt_0.1.1.model.tar.gz', From cd6275e3443b4c385c148d76b5b3d2c4b43be1af Mon Sep 17 00:00:00 2001 From: huangyuxin Date: Sun, 24 Apr 2022 06:53:50 +0000 Subject: [PATCH 4/4] add test for long audio in cli, test=doc --- tests/unit/cli/test_cli.sh | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/tests/unit/cli/test_cli.sh b/tests/unit/cli/test_cli.sh index 87c24b099ce..96e1e02357a 100755 --- a/tests/unit/cli/test_cli.sh +++ b/tests/unit/cli/test_cli.sh @@ -1,5 +1,6 @@ #!/bin/bash set -e +echo -e "\e[1;31monly if you see 'Test success !!!', the cli testing is successful\e[0m" # Audio classification wget -c https://paddlespeech.bj.bcebos.com/PaddleAudio/cat.wav https://paddlespeech.bj.bcebos.com/PaddleAudio/dog.wav @@ -13,6 +14,13 @@ wget -c https://paddlespeech.bj.bcebos.com/PaddleAudio/zh.wav https://paddlespee paddlespeech asr --input ./zh.wav paddlespeech asr --model transformer_librispeech --lang en --input ./en.wav +# long audio restriction +wget -c wget https://paddlespeech.bj.bcebos.com/datasets/single_wav/zh/test_long_audio_01.wav +paddlespeech asr --input test_long_audio_01.wav +if [ $? -ne 1 ]; then + exit 1 +fi + # Text To Speech paddlespeech tts --input "你好,欢迎使用百度飞桨深度学习框架!" paddlespeech tts --am speedyspeech_csmsc --input "你好,欢迎使用百度飞桨深度学习框架!" @@ -56,3 +64,6 @@ paddlespeech stats --task cls paddlespeech stats --task text paddlespeech stats --task vector paddlespeech stats --task st + + +echo "Test success !!!"