通过netron软件查看结构: 64288=392*164,说明model维度没有问题 剪枝模型存储为预测模型时: checkpoint没有报错,说明参数也没有问题
可以看到输出的channel和参数并不匹配,请确认模型是否有问题:
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
利用c++ 预测接口,采取fp32预测方式可以正常运行,采取tensorrt fp32方式报错:
通过netron软件查看结构:
64288=392*164,说明model维度没有问题

剪枝模型存储为预测模型时:
checkpoint没有报错,说明参数也没有问题