[TensorRT] ERROR: Network must have at least one output

[TensorRT] ERROR: Network must have at least one output

yolov3 转 tensorrt,运行onnx转tensorrt 有时会遇到上述错误。
onnx转tensorrt.py文件中,在parser.parse下with循环外添加两行代码

last_layer = network.get_layer(network.num_layers - 1)
network.mark_output(last_layer.get_output(0))
[TensorRT] ERROR: Network must have at least one output
[TensorRT] ERROR: Network must have at least one output