Gridsampler to TensorRT

[jyang68sh]

Hi I would like to export a retrained model from onnx to tensorrt on resolution size: 2048x1024, 1024x512.

However, gridsampler is not a supported op by onnx.

Could you give instruction on how to export onnx to tensorrt?

Thanks! Great work by the way.

[jyang68sh]

@lxtGH Hello. Could you please give some instruction?

[gaowq2017]

@jyang68sh
You can refer the link
That is torch.onnx.export(....., opset_version=16, ......) Thanks

[lxtGH]

@jyang68sh Hi! We use opset =16 . The grid sampler is build from mmcv. Thanks !!!