- 使用TensorRT 6.0.1 OnnxParser parse错误为
In node 0 (importModel): INVALID_GRAPH: Assertion failed: tensors.count(input_name) - 使用TensorRT源码编译后的onnx2trt 命令行转换,没有报错
没法找到具体原因
-
jamesguo |
onnx 转 tensorrt出错
-
刘看山 |
@wjj19950828 升级onnx-tensorrt可以解决这个问题
-
hello 请问你这个问题是如何解决的呢 我目前也遇到了相似情况
-
刘看山 |
-
刘看山 |
onnx图没有按照标准的格式存储。
count应该是每个tensorrt必须的字段。现在onnx-tensorrt还无法推理默认的尺寸。
可以先check一下onnx是否valid,方式为使用torch.onnx
load模型check一下格式。