mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-06 15:44:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor adjustments

Browse Source
This commit is contained in:
jinzr 2023-12-13 10:25:12 +08:00
parent 71abe27d0e
commit 23f3b15c9d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/zipformer/export-onnx-streaming-ctc.py
View File

@ -542,7 +542,7 @@ def main():
opset_version = 13 opset_version = 13
logging.info("Exporting model") logging.info("Exporting model")
model_filename = params.exp_dir / f"ctc-streaming-encoder-{suffix}.onnx" model_filename = params.exp_dir / f"ctc-{suffix}.onnx"
export_streaming_ctc_model_onnx( export_streaming_ctc_model_onnx(
model, model,
model_filename, model_filename,
@ -555,7 +555,7 @@ def main():
logging.info("Generate int8 quantization models") logging.info("Generate int8 quantization models")
model_filename_int8 = params.exp_dir / f"ctc-streaming-encoder-{suffix}.int8.onnx" model_filename_int8 = params.exp_dir / f"ctc-{suffix}.int8.onnx"
quantize_dynamic( quantize_dynamic(
model_input=model_filename, model_input=model_filename,
model_output=model_filename_int8, model_output=model_filename_int8,