diff --git a/.github/scripts/test-online-transducer.sh b/.github/scripts/test-online-transducer.sh index 2c2c3815..f2f94530 100755 --- a/.github/scripts/test-online-transducer.sh +++ b/.github/scripts/test-online-transducer.sh @@ -18,7 +18,6 @@ log "Run LSTM transducer (English)" log "------------------------------------------------------------" repo_url=https://huggingface.co/csukuangfj/sherpa-onnx-lstm-en-2023-02-17 - log "Start testing ${repo_url}" repo=$(basename $repo_url) log "Download pretrained model and test-data from $repo_url" @@ -45,3 +44,101 @@ for wave in ${waves[@]}; do done rm -rf $repo + +log "------------------------------------------------------------" +log "Run LSTM transducer (Chinese)" +log "------------------------------------------------------------" + +repo_url=https://huggingface.co/csukuangfj/sherpa-onnx-lstm-zh-2023-02-20 +log "Start testing ${repo_url}" +repo=$(basename $repo_url) +log "Download pretrained model and test-data from $repo_url" + +GIT_LFS_SKIP_SMUDGE=1 git clone $repo_url +pushd $repo +git lfs pull --include "*.onnx" +popd + +waves=( +$repo/test_wavs/0.wav +$repo/test_wavs/1.wav +$repo/test_wavs/2.wav +) + +for wave in ${waves[@]}; do + time $EXE \ + $repo/tokens.txt \ + $repo/encoder-epoch-11-avg-1.onnx \ + $repo/decoder-epoch-11-avg-1.onnx \ + $repo/joiner-epoch-11-avg-1.onnx \ + $wave \ + 4 +done + +rm -rf $repo + +log "------------------------------------------------------------" +log "Run streaming Zipformer transducer (English)" +log "------------------------------------------------------------" + +repo_url=https://huggingface.co/csukuangfj/sherpa-onnx-streaming-zipformer-en-2023-02-21 +log "Start testing ${repo_url}" +repo=$(basename $repo_url) +log "Download pretrained model and test-data from $repo_url" + +GIT_LFS_SKIP_SMUDGE=1 git clone $repo_url +pushd $repo +git lfs pull --include "*.onnx" +popd + +waves=( +$repo/test_wavs/1089-134686-0001.wav +$repo/test_wavs/1221-135766-0001.wav +$repo/test_wavs/1221-135766-0002.wav +) + +for wave in ${waves[@]}; do + time $EXE \ + $repo/tokens.txt \ + $repo/encoder-epoch-99-avg-1.onnx \ + $repo/decoder-epoch-99-avg-1.onnx \ + $repo/joiner-epoch-99-avg-1.onnx \ + $wave \ + 4 +done + +rm -rf $repo + +log "------------------------------------------------------------" +log "Run streaming Zipformer transducer (Bilingual, Chinse + English)" +log "------------------------------------------------------------" + +repo_url=https://huggingface.co/csukuangfj/sherpa-onnx-streaming-zipformer-bilingual-zh-en-2023-02-20 +log "Start testing ${repo_url}" +repo=$(basename $repo_url) +log "Download pretrained model and test-data from $repo_url" + +GIT_LFS_SKIP_SMUDGE=1 git clone $repo_url +pushd $repo +git lfs pull --include "*.onnx" +popd + +waves=( +$repo/test_wavs/0.wav +$repo/test_wavs/1.wav +$repo/test_wavs/2.wav +$repo/test_wavs/3.wav +$repo/test_wavs/4.wav +) + +for wave in ${waves[@]}; do + time $EXE \ + $repo/tokens.txt \ + $repo/encoder-epoch-99-avg-1.onnx \ + $repo/decoder-epoch-99-avg-1.onnx \ + $repo/joiner-epoch-99-avg-1.onnx \ + $wave \ + 4 +done + +rm -rf $repo