WARNING: ONNX model has a newer ir_version (0.0.4) than this parser was built against (0.0.3).
While parsing node number 0 [Conv]:
ERROR: ModelImporter.cpp:296 In function importModel:
[5] Assertion failed: tensors.count(input_name)
I think this error comes from the ONNX parser having an issue parsing your graph, specifically that it found a node with no input I believe.
There are a few things you can try for this:
It seems that PyTorch 1.3 is a little more up to date than the current parsers (opset 11), so that may be the root cause for now. You can try downgrading to 1.2 just for exporting your Pytorch to ONNX, see here: https://github.com/onnx/onnx-tensorrt/issues/302
You can try inspecting your graph with the ONNX python library to see if it gives you any hints to the issue:
Update: This works on PyTorch 1.2.0 - downgrading to that should work. However, if you’re looking to do any upsampling op as part of your export to ONNX, you’re going to have some trouble.
Any more recent updates on this? I’m seeing the same problem, I need to use pytorch 1.4 for an unrelated reason but building a trt engine from the resulting onnx files fails because of this error.