使用TensorRT3.0转换基于tensorflow的resnet-50网络

时间:2018-07-13 09:22:36

标签: python tensorflow tensorrt

看到这个问题

  

回溯(最近通话最近):   在import_graph_def的第489行中,文件“ /root/anaconda3/lib/python3.4/site-packages/tensorflow/python/framework/importer.py”   graph._c_graph,序列化,选项)#pylint:disable = protected-access   tensorflow.python.framework.errors_impl.InvalidArgumentError:形状必须为2级,但对于'import / dense_p7 / MatMul'(op:'MatMul')的输入形状为[1,256,1,1],[256,1则为4级]

我们可以成功完成tensorrt子图的约定,但是我们在推理阶段遇到了问题。我的模型是基于resnet-50的张量流。谁能帮助我解决这个问题,谢谢!

0 个答案:

没有答案