├── LICENSE ├── README.md ├── dynamic_shape_example.py ├── helpers ├── __init__.py ├── trt_helper.py └── trt_int8_calibration_helper.py ├── model_128.onnx ├── pytorch_onnx_trt.ipynb ├── pytorch_onnx_trt.pdf └── trt_int8_demo.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RizhaoCai/PyTorch_ONNX_TensorRT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RizhaoCai/PyTorch_ONNX_TensorRT/HEAD/README.md -------------------------------------------------------------------------------- /dynamic_shape_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RizhaoCai/PyTorch_ONNX_TensorRT/HEAD/dynamic_shape_example.py -------------------------------------------------------------------------------- /helpers/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /helpers/trt_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RizhaoCai/PyTorch_ONNX_TensorRT/HEAD/helpers/trt_helper.py -------------------------------------------------------------------------------- /helpers/trt_int8_calibration_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RizhaoCai/PyTorch_ONNX_TensorRT/HEAD/helpers/trt_int8_calibration_helper.py -------------------------------------------------------------------------------- /model_128.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RizhaoCai/PyTorch_ONNX_TensorRT/HEAD/model_128.onnx -------------------------------------------------------------------------------- /pytorch_onnx_trt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RizhaoCai/PyTorch_ONNX_TensorRT/HEAD/pytorch_onnx_trt.ipynb -------------------------------------------------------------------------------- /pytorch_onnx_trt.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RizhaoCai/PyTorch_ONNX_TensorRT/HEAD/pytorch_onnx_trt.pdf -------------------------------------------------------------------------------- /trt_int8_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RizhaoCai/PyTorch_ONNX_TensorRT/HEAD/trt_int8_demo.py --------------------------------------------------------------------------------