Hi,
I am exporting model using torchtotrt developed by NVIDIA please find the link below for the conversion

  
      
      GitHub
  
  
    

NVIDIA-AI-IOT/torch2trt 20

An easy to use PyTorch to TensorRT converter. Contribute to NVIDIA-AI-IOT/torch2trt development by creating an account on GitHub.