Description
I’m going to transform a onnx model into TensorRT engine file. Tensorrt provides a sample for helping us to define a custom layer. However, this sample was written in caffe. I couldn’t find any samples or docs related to pytorch.
How to define a custom plugin in TensorRT7? any cues would be highly appreciated
Environment
TensorRT Version: 7.0
GPU Type: 1080Ti
Nvidia Driver Version:
CUDA Version: 10.2
CUDNN Version: 7.6
Operating System + Version: Ubuntu18.04
Python Version (if applicable): 3.7
TensorFlow Version (if applicable):
PyTorch Version (if applicable): 1.3
Baremetal or Container (if container which image + tag):