/pytorch-to-tensorrt-model-converter

This project provides a comprehensive Python script to convert a PyTorch model to an ONNX model and then to a TensorRT engine for NVIDIA GPUs, followed by performing inference using the TensorRT engine. This script is designed to handle the entire conversion process seamlessly.

Primary LanguagePythonApache License 2.0Apache-2.0

Stargazers