From 23f3b15c9d1fadc0184632ab0f0cf9b4f7d5537f Mon Sep 17 00:00:00 2001 From: jinzr Date: Wed, 13 Dec 2023 10:25:12 +0800 Subject: [PATCH] minor adjustments --- egs/librispeech/ASR/zipformer/export-onnx-streaming-ctc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/zipformer/export-onnx-streaming-ctc.py b/egs/librispeech/ASR/zipformer/export-onnx-streaming-ctc.py index dac89b5f3..201019649 100755 --- a/egs/librispeech/ASR/zipformer/export-onnx-streaming-ctc.py +++ b/egs/librispeech/ASR/zipformer/export-onnx-streaming-ctc.py @@ -542,7 +542,7 @@ def main(): opset_version = 13 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( model, model_filename, @@ -555,7 +555,7 @@ def main(): 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( model_input=model_filename, model_output=model_filename_int8,