New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
paddle 1.8 ernie生成文本分类模型后,使用paddle inference tensorRT库报错 #646
Comments
用2.0.5版本跑,python3.6 |
非 TRT 的情况下, 能够正常运行吗? |
@OleNet 可以 |
请问2.0.5是什么版本?paddle好像没有发布这个版本, |
@shangzhizhou我typo了,应该是 paddlepaddle_gpu-2.0.2 |
好的,我们尽快复现一下问题 |
您好,这边根据提供的模型和脚本进行了测试,目前发现脚本存在一些问题,主要是设置trt子图的动态输入理解有误,下面是修改后可以运行的代码,参考代码链接 |
十分感谢。 |
This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Feel free to reopen it. Thank you for your contributions. |
docker环境
docker pull hub.baidubce.com/paddlepaddle/paddle:1.8.0-gpu-cuda10.0-cudnn7-trt6
python2.7 / ernie v2.4.0
操作步骤:
运行 finetune_classifier_dygraph.py,得到model后,使用paddle_inference_demo中的例子做推理加速。
错误信息:
Error Message Summary:
Error: Operator skip_layernorm has not been registered
[Hint: op_info_ptr should not be null.] at (/paddle /trt_refine_int8/paddle/fluid/framework/op_info.h:140)
相关代码&模型:
链接: https://pan.baidu.com/s/1xVIyc2ntboVq_f7hse4uyg 提取码: gudn 复制这段内容后打开百度网盘手机App,操作更方便哦
The text was updated successfully, but these errors were encountered: