bnabis93 / tensorrt-toy

tensorrt-toy code

TODO

1. TensorRT Cifar10 PTQ example
2. TensorRT Cifar10 QAT example
3. Torch-TensorRT basic usage example
4. Torch-TensorRT vs Torch -> onnx -> TensorRT benchmark

Tensorrt toycode

• TensorRT Tips
• Simple convert example for torch model to trt
• Torch vs TensorRT benchmark using cifar10 dataset and resnet model
• Torch-tensorrt basic usage