VertexC/dl-infer-perf

onnx->trt fail on mobilenet

Opened this issue · 2 comments

VertexC commented 
root@ubuntu1804-lts-base:/scratch/dev/dev/dl-infer-perf# python3 infer_perf/onnx2trt.py mobilenet
[TensorRT] ERROR: Network must have at least one output
Traceback (most recent call last):
  File "infer_perf/onnx2trt.py", line 80, in <module>
    runner = onnx2trt_runner(args.model, batch_size=args.batch)
  File "infer_perf/onnx2trt.py", line 60, in onnx2trt_runner
    context = engine.create_execution_context()
AttributeError: 'NoneType' object has no attribute 'create_execution_context'
VertexC commented

The error has something to do onnx's model version and current trt verison. Need further investigation.