-
项目“朝圣者”提供的torch2trt工具,旨在将PyTorch模型高效转换为TensorRT引擎,适用于C/C++开发环境
资源介绍
该项目由torch2trt和torch2trt_dynamic分叉,此项目的目的是提供一种将pytorch模型直接转换为TensorRT引擎的朝圣者项目。 TensorRT引擎。
如果您的模型是基于pytorch构建的,则可以加快您的项目的工作速度;我们已经测试了多个模型,这些模型都可以正常工作,并且我们将添加更多测试和插件来支持复杂模型的模型。
我们不希望使用onnx的原因是:onnx是另一种不太必要的中间件;
维护onnx插件并不容易