0

看到这个问题

Traceback(最近一次调用最后一次):文件“/root/anaconda3/lib/python3.4/site-packages/tensorflow/python/framework/importer.py”,第 489 行,在 import_graph_def graph._c_graph,序列化,选项)# pylint: disable=protected-access tensorflow.python.framework.errors_impl.InvalidArgumentError: Shape must be rank 2 but is rank 4 for 'import/dense_p7/MatMul' (op: 'MatMul') with input shapes: [1,256,1, 1],[256,1]

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

4

0 回答 0