/pytorch-infer-utils

This package proposes simplified exporting pytorch models to ONNX and TensorRT, and also gives some base interface for ML models inference.

Primary LanguagePythonApache License 2.0Apache-2.0

Stargazers