From de1880948b44d25437510c15d52cef1a32bc0e6e Mon Sep 17 00:00:00 2001 From: Jingzhao Ou Date: Thu, 11 May 2023 17:37:24 -0700 Subject: [PATCH] Added streaming conformer test (#150) --- .github/scripts/test-online-transducer.sh | 43 +++++++++++++++++++++++ 1 file changed, 43 insertions(+) diff --git a/.github/scripts/test-online-transducer.sh b/.github/scripts/test-online-transducer.sh index 2a5bac87..582eda79 100755 --- a/.github/scripts/test-online-transducer.sh +++ b/.github/scripts/test-online-transducer.sh @@ -209,3 +209,46 @@ if [ $EXE == "sherpa-onnx-ffmpeg" ]; then fi rm -rf $repo + +log "------------------------------------------------------------" +log "Run streaming Conformer transducer (English)" +log "------------------------------------------------------------" + +repo_url=https://huggingface.co/csukuangfj/sherpa-onnx-streaming-conformer-en-2023-05-09 +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" +ls -lh *.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-99-avg-1.onnx \ + $repo/decoder-epoch-99-avg-1.onnx \ + $repo/joiner-epoch-99-avg-1.onnx \ + $wave \ + 2 +done + +for wave in ${waves[@]}; do + time $EXE \ + $repo/tokens.txt \ + $repo/encoder-epoch-99-avg-1.int8.onnx \ + $repo/decoder-epoch-99-avg-1.int8.onnx \ + $repo/joiner-epoch-99-avg-1.int8.onnx \ + $wave \ + 2 +done + +rm -rf $repo