Assertion failed: tensors.count(input_name)

场景

PyTorch 1.3，TensorRT 6.0，ONNX 1.5

PyTorch 训练好的 CRNN 模型，转换为 ONNX 之后无法转为 TensorRT，出现

----------------------------------------------------------------
Input filename:   crnn.onnx
ONNX IR version:  0.0.4
Opset version:    9
Producer name:    pytorch
Producer version: 1.3
Domain:
Model version:    0
Doc string:
----------------------------------------------------------------
Parsing model
While parsing node number 50 [GRU -> "269"]:
ERROR: /opt/onnx-tensorrt/ModelImporter.cpp:487 In function importModel:
[5] Assertion failed: tensors.count(input_name)

原因

https://github.com/onnx/onnx-tensorrt/blob/master/operators.md

↑ onnx-tensorrt 支持的操作列表，根本就不支持 RNN，包括 LSTM、GRU 等都不支持

解决