MASR icon indicating copy to clipboard operation
MASR copied to clipboard

PUNC模型怎么跑onnx推断

Open DrewdropLife opened this issue 9 months ago • 1 comments

https://github.com/yeyupiaoling/MASR/blob/4ee83b1e71d29e602c3aa93ec90c2761abb80cc8/masr/infer_utils/punc_predictor.py#L21 这个模型要怎么转成onnx? 而且转完之后,在3.0版本中怎么运行onnx推断呢?压缩包里缺失相应的configs.yaml、jieba_user_dict、tokens.json(这个我在2.x版本中找到了)。

我是通过如下命令转的onnx,但显然完全错误: paddle2onnx --model_dir "pun_models" --model_filename "model.pdmodel" --params_filename model.pdiparams --save_file model.onnx 然后得到的onnx模型的输入名是input_ids和token_type_ids。 应该怎么转呀?

DrewdropLife avatar Mar 20 '25 07:03 DrewdropLife

@DrewdropLife 使用的是不同的框架模型,所以不能混用

yeyupiaoling avatar Apr 28 '25 12:04 yeyupiaoling