yolov3 prune预测模型tensorrt 预测报错
Created by: barry2025
- 版本、环境信息： 1）PaddlePaddle版本：1.6.1 2）GPU：cuda10.1和cuda9 -预测信息 1）C++预测：version.txt GIT COMMIT ID: 70c073a0 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方式报错：
通过netron软件查看结构：
64288=392*164，说明model维度没有问题
剪枝模型存储为预测模型时：
checkpoint没有报错，说明参数也没有问题