TorchRT provides a simple way to convert a PyTorch nn.Module to a TensorRT-equipped module. While the goal of this package is similar to that of torch2trt from NVIDIA, TorchRT does this by first converting the module to ONNX, allowing options to optimize the model graph without the need to maintain conversion code for each operator.
-
Notifications
You must be signed in to change notification settings - Fork 1
Simple tool for PyTorch >> ONNX >> TensorRT conversion
License
jinyeom/torchrt
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Simple tool for PyTorch >> ONNX >> TensorRT conversion
Topics
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published