论坛
BBS
会员中心
Space
用户
登录
用户名
Email
自动登录
找回密码
密码
登录
立即注册
搜索
搜索
本版
帖子
用户
本版
帖子
用户
CUDA
技术
工具
资源
入门
社区
论坛
开启辅助访问
切换到宽版
好友
帖子
收藏
道具
勋章
任务
设置
我的收藏
退出
腾讯QQ
微信登录
论坛
›
AI 和数据科学
›
SDK
›
查看内容
发表新帖
查看:
7076
|
回复:
1
同一个模型,TensorRT7.1.3输出结果错误,TensorRT7.2.1输出结果...
WangCanlong
WangCanlong
当前离线
WangCanlong
初入江湖
|
显示全部楼层
|
阅读模式
1
#
电梯直达
发表于 2021-2-1 19:50:43
7076
1
项目地址:github。coral-cnn,模型afad-coral。
onnxruntime输出结果正确,使用TensorRT7.1.3版本onnx转trt推理结果完全错误,换成TensorRT7.2.1版本则输出正确结果。转换过程中无报错,数据预处理完全一致,唯一的区别就是trt的版本,想知道是什么原因造成的trt7.1.3结果错误,能否解决?。
使用道具
举报
回复
WangCanlong
当前离线
WangCanlong
初入江湖
|
显示全部楼层
2
#
发表于 2021-2-2 14:02:58
已解决。应该是TensorRT7.1.3对nn.Parameter的支持不够完善,模型训练完之后这就是个固定参数,可以放到engine外面做,这样该模型在trt7.1.3版本也能正常使用了。
使用道具
举报
回复
支持
反对
高级模式
B
Color
Image
Link
Quote
Code
Smilies
您需要登录后才可以回帖
登录
|
立即注册
发表回复