gorodnitskiy / pytorch-infer-utils

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

Related projects: