Add ReazonSpeech Japanese pre-trained model (#1203)

This commit is contained in:
Fangjun Kuang
2024-08-02 10:21:24 +08:00
committed by GitHub
parent 53484fcd9b
commit 35c1b4a7a9
5 changed files with 96 additions and 18 deletions

View File

@@ -15,6 +15,50 @@ echo "PATH: $PATH"
which $EXE
log "------------------------------------------------------------------------"
log "Run zipformer transducer models (Japanese from ReazonSpeech) "
log "------------------------------------------------------------------------"
url=https://github.com/k2-fsa/sherpa-onnx/releases/download/asr-models/sherpa-onnx-zipformer-ja-reazonspeech-2024-08-01.tar.bz2
name=$(basename $url)
curl -SL -O $url
tar xvf $name
rm $name
repo=$(basename -s .tar.bz2 $name)
ls -lh $repo
cat $repo/test_wavs/*.txt
log "test $repo"
test_wavs=(
1.wav
2.wav
3.wav
4.wav
5.wav
)
for w in ${test_wavs[@]}; do
time $EXE \
--tokens=$repo/tokens.txt \
--encoder=$repo/encoder-epoch-99-avg-1.onnx \
--decoder=$repo/decoder-epoch-99-avg-1.onnx \
--joiner=$repo/joiner-epoch-99-avg-1.onnx \
--debug=1 \
$repo/test_wavs/$w
done
for w in ${test_wavs[@]}; do
time $EXE \
--tokens=$repo/tokens.txt \
--encoder=$repo/encoder-epoch-99-avg-1.int8.onnx \
--decoder=$repo/decoder-epoch-99-avg-1.onnx \
--joiner=$repo/joiner-epoch-99-avg-1.int8.onnx \
--debug=1 \
$repo/test_wavs/$w
done
rm -rf $repo
log "------------------------------------------------------------------------"
log "Run Nemo fast conformer hybrid transducer ctc models (transducer branch)"
log "------------------------------------------------------------------------"