首页 Paddle框架 帖子详情
yolov3 prune预测模型tensorrt 预测报错
收藏
快速回复
Paddle框架 问答深度学习模型训练炼丹技巧 909 1
yolov3 prune预测模型tensorrt 预测报错
收藏
快速回复
Paddle框架 问答深度学习模型训练炼丹技巧 909 1
  • 版本、环境信息:
       1)PaddlePaddle版本:1.6.1
       2)GPU:cuda10.1和cuda9
    -预测信息
       1)C++预测:version.txt
    GIT COMMIT ID: 70c073a
    WITH_MKL: ON
    WITH_MKLDNN: OFF
    WITH_GPU: ON
    CUDA version: 10.1
    CUDNN version: v7 
  • 复现信息:剪枝预测模型地址 https://bj.bcebos.com/v1/mct-imagedb/tangxin-bos/yolov3_r34_prune.tar.gz
  • 问题描述:
    利用c++ 预测接口,采取fp32预测方式可以正常运行,采取tensorrt fp32方式报错:
    image

通过netron软件查看结构:
image 64288=392*164,说明model维度没有问题
剪枝模型存储为预测模型时:
image
checkpoint没有报错,说明参数也没有问题

0
收藏
回复
全部评论(1)
时间顺序
Zhaolong Xing
#2 回复于2020-02

可以看到输出的channel和参数并不匹配,请确认模型是否有问题:

image

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户