ONNX -> TRT fails

[gan3sh500]

trafficstars

The exported onnx model from this repo currently fails with TRT 7.x.

The failure happens when the onnx model is parsed by TRT. The parsed model will be missing most of its layers.

I was able to fix this by changing the opset_version in torch.onnx.export. I can PR if you see this issue as relevant. I also have scripts to convert the ONNX model to TRT engine without using onnx2trt library which is pretty difficult to use especially if using Jetson.

[SthPhoenix]

UPD: Found it, would try it later, thanks!

Hi! Can you share ONNX to TRT conversion example? I was able to convert model to TRT, but outputs are of completely different shapes

[gan3sh500]

@SthPhoenix Ah yes. Outputs are flatten. Should reshape back.

[yumaofan]

Please refer #116