用户
 找回密码
 立即注册
发表于 2021-2-1 19:50:43
70761
项目地址:github。coral-cnn,模型afad-coral。
onnxruntime输出结果正确,使用TensorRT7.1.3版本onnx转trt推理结果完全错误,换成TensorRT7.2.1版本则输出正确结果。转换过程中无报错,数据预处理完全一致,唯一的区别就是trt的版本,想知道是什么原因造成的trt7.1.3结果错误,能否解决?。
使用道具 举报 回复
发表于 2021-2-2 14:02:58
已解决。应该是TensorRT7.1.3对nn.Parameter的支持不够完善,模型训练完之后这就是个固定参数,可以放到engine外面做,这样该模型在trt7.1.3版本也能正常使用了。
使用道具 举报 回复 支持 反对
发新帖
您需要登录后才可以回帖 登录 | 立即注册