找回密码
 立即注册
meiyaya 该用户已被删除
发表于 2019-3-21 15:09:21
9100
环境:tensorRT 5.0
python 3.5
tensorflow 1.8
思路:使用tf的.pb model转化为 .uff,最后进行推理优化。
问题,生成推理引擎的时候,总是提示uff解析失败
文本就是一个如长度为60个字符的数据,想的模拟图像的CHW输入,将其变为(60,1,1)的形式,UFF文件可以生成。
错误如下:
[TensorRT] ERROR: Parameter check failed at : ../builder/Network.cpp::addInput::406, condition: isValidDims(dims)
[TensorRT] ERROR: UFFParser: Failed to pasrseInput for node input_1
请大家帮忙看一下啊。TRT有文本预处理的这块操作,首先要将文本变为tensor吗,哪里有这块的相关资料呢
使用道具 举报 回复
发新帖
您需要登录后才可以回帖 登录 | 立即注册

zzczczxczxczx