/torch2onnx2trt

A simple package that wraps PyTorch models conversion to ONNX and TensorRT

Primary LanguagePythonMIT LicenseMIT

Stargazers