PaddleClas icon indicating copy to clipboard operation
PaddleClas copied to clipboard

Published 20 hours ago • verified publisher icon PaddlePaddle

Efficientnet是否支持tensorrt推理

Open litao-zhx opened this issue 2 years ago • 8 comments

litao-zhx avatar May 18 '22 08:05 litao-zhx

理论上是支持的，遇到bug了吗

RainFrost1 avatar May 19 '22 07:05 RainFrost1

对的，和pytorch版用tensorrt推理一样，觉得是因为最后那层的全局平均池化所致

litao-zhx avatar May 20 '22 09:05 litao-zhx

1653274745(1) 完整报错如上

litao-zhx avatar May 23 '22 02:05 litao-zhx

好的，我验证一下

RainFrost1 avatar May 30 '22 09:05 RainFrost1

你好，请问问题有找到吗？

litao-zhx avatar Jun 15 '22 03:06 litao-zhx

我这边测试是可以跑通的，主要要安装好合适的trt的环境，相应的库见链接

RainFrost1 avatar Jun 28 '22 07:06 RainFrost1

我的环境就是根据这个链接去安装的，您那边测试时的输入尺寸是多少

litao-zhx avatar Jun 28 '22 07:06 litao-zhx

yaml文件中默认的尺寸

RainFrost1 avatar Jul 11 '22 11:07 RainFrost1