/torchtrtz

Pytorch models built and saved to generate tensorrt weights file. Sibling project to use the generated trt weights file and create tensorrt engine using TensorRT C++ api.

Primary LanguagePython

Stargazers