mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-19 05:54:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor fixes

Browse Source
This commit is contained in:
JinZr 2023-08-12 15:58:17 +08:00
parent 2f4a0fd9fd
commit 5baa2be81f
3 changed files with 11 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
.github/scripts/run-wenetspeech-pruned-transducer-stateless2.sh vendored
View File

@ -28,7 +28,7 @@ popd
log "Test exporting to ONNX format" log "Test exporting to ONNX format"
./pruned_transducer_stateless2/export.py \ ./pruned_transducer_stateless2/export-onnx.py \
--exp-dir $repo/exp \ --exp-dir $repo/exp \
--lang-dir $repo/data/lang_char \ --lang-dir $repo/data/lang_char \
--epoch 99 \ --epoch 99 \
@ -57,19 +57,17 @@ log "Decode with ONNX models"
./pruned_transducer_stateless2/onnx_check.py \ ./pruned_transducer_stateless2/onnx_check.py \
--jit-filename $repo/exp/cpu_jit.pt \ --jit-filename $repo/exp/cpu_jit.pt \
--onnx-encoder-filename $repo/exp/encoder.onnx \ --onnx-encoder-filename $repo/exp/encoder-epoch-10-avg-2.onnx \
--onnx-decoder-filename $repo/exp/decoder.onnx \ --onnx-decoder-filename $repo/exp/decoder-epoch-10-avg-2.onnx \
--onnx-joiner-filename $repo/exp/joiner.onnx \ --onnx-joiner-filename $repo/exp/joiner-epoch-10-avg-2.onnx \
--onnx-joiner-encoder-proj-filename $repo/exp/joiner_encoder_proj.onnx \ --onnx-joiner-encoder-proj-filename $repo/exp/joiner_encoder_proj-epoch-10-avg-2.onnx \
--onnx-joiner-decoder-proj-filename $repo/exp/joiner_decoder_proj.onnx --onnx-joiner-decoder-proj-filename $repo/exp/joiner_decoder_proj-epoch-10-avg-2.onnx
./pruned_transducer_stateless2/onnx_pretrained.py \ ./pruned_transducer_stateless2/onnx_pretrained.py \
--tokens $repo/data/lang_char/tokens.txt \ --tokens $repo/data/lang_char/tokens.txt \
--encoder-model-filename $repo/exp/encoder.onnx \ --encoder-model-filename $repo/exp/encoder-epoch-99-avg-1.onnx \
--decoder-model-filename $repo/exp/decoder.onnx \ --decoder-model-filename $repo/exp/decoder-epoch-99-avg-1.onnx \
--joiner-model-filename $repo/exp/joiner.onnx \ --joiner-model-filename $repo/exp/joiner-epoch-99-avg-1.onnx \
--joiner-encoder-proj-model-filename $repo/exp/joiner_encoder_proj.onnx \
--joiner-decoder-proj-model-filename $repo/exp/joiner_decoder_proj.onnx \
$repo/test_wavs/DEV_T0000000000.wav \ $repo/test_wavs/DEV_T0000000000.wav \
$repo/test_wavs/DEV_T0000000001.wav \ $repo/test_wavs/DEV_T0000000001.wav \
$repo/test_wavs/DEV_T0000000002.wav $repo/test_wavs/DEV_T0000000002.wav

2
egs/wenetspeech/ASR/pruned_transducer_stateless2/export-onnx.py
View File

@ -498,7 +498,7 @@ def main():
quantize_dynamic( quantize_dynamic(
model_input=decoder_filename, model_input=decoder_filename,
model_output=decoder_filename_int8, model_output=decoder_filename_int8,
op_types_to_quantize=["MatMul"], op_types_to_quantize=["MatMul", "Gather"],
weight_type=QuantType.QInt8, weight_type=QuantType.QInt8,
) )

2
egs/wenetspeech/ASR/pruned_transducer_stateless2/pretrained.py
View File

@ -320,7 +320,7 @@ def main():
s = "\n" s = "\n"
for filename, hyp in zip(params.sound_files, hyps): for filename, hyp in zip(params.sound_files, hyps):
words = " ".join(hyp) words = "".join(hyp)
s += f"{filename}:\n{words}\n\n" s += f"{filename}:\n{words}\n\n"
logging.info(s) logging.info(s)