From bf918fe311aa1271abcdabf0157a53b71b45906e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=B9=9D=E8=80=B3?= Date: Wed, 29 Mar 2023 17:17:12 +0800 Subject: [PATCH] general punc model conversion onnx, fix bug --- funasr/export/test/test_onnx_punc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/funasr/export/test/test_onnx_punc.py b/funasr/export/test/test_onnx_punc.py index dc63176fb..62689a904 100644 --- a/funasr/export/test/test_onnx_punc.py +++ b/funasr/export/test/test_onnx_punc.py @@ -3,7 +3,7 @@ import numpy as np if __name__ == '__main__': - onnx_path = "/disk1/mengzhe.cmz/workspace/FunASR/funasr/export/damo/punc_ct-transformer_zh-cn-common-vocab272727-pytorch/model.onnx" + onnx_path = "../damo/punc_ct-transformer_zh-cn-common-vocab272727-pytorch/model.onnx" sess = onnxruntime.InferenceSession(onnx_path) input_name = [nd.name for nd in sess.get_inputs()] output_name = [nd.name for nd in sess.get_outputs()]