看到这个问题
回溯(最近通话最近): 在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的张量流。谁能帮助我解决这个问题,谢谢!