文件名称：torch2trt 是一个易于使用的PyTorch到TensorRT转换器-python
文件大小：162KB
文件格式：ZIP
更新时间：2024-07-08 12:21:38
机器学习
torch2trt 是一个易于使用的PyTorch到TensorRT转换器，它使用TensorRT Python API实现 torch2trt torch2trt 是 PyTorch 到 TensorRT 的转换器，它利用了 TensorRT Python API。 转换器易于使用 - 使用单个函数调用 torch2trt 转换模块 易于扩展 - 用 Python 编写您自己的层转换器并使用 @tensorrt_converter 注册 如果您发现问题，请告诉我们！ 请注意，此转换器对 TensorRT / PyTorch 的覆盖范围有限。 我们创建它主要是为了轻松优化 JetBot 项目中使用的模型。 如果您发现转换器对其他型号有帮助，请告诉我们。 用法 下面是一些用法示例，更多信息请查看 notebook。 转换 import torch from torch2trt import torch2trt from torchvision.models.alexnet import alexnet # 创建一些常规的 pytorch 模型... model = alexn