文件名称:朝圣者项目:torch2trt,将pytorch模型快速转换为TensorRT引擎。-C/C++开发
文件大小:345KB
文件格式:ZIP
更新时间:2024-06-15 20:34:44
C/C++ Machine Learning
该项目由torch2trt和torch2trt_dynamic分叉,此项目的目的是提供一种将pytorch模型直接转换为TensorRT引擎的朝圣者项目。 TensorRT引擎。 如果您的模型是基于pytorch构建的,则可以加快您的项目的工作速度;我们已经测试了多个模型,这些模型都可以正常工作,并且我们将添加更多测试和插件来支持复杂模型的模型。 我们不希望使用onnx的原因是:onnx是另一种不太必要的中间件; 维护onnx插件并不容易