文件名称：torch2trt：易于使用的PyTorch到TensorRT转换器
文件大小：155KB
文件格式：ZIP
更新时间：2024-03-17 02:08:55
inference pytorch classification tensorrt jetson-tx2
火炬2trt torch2trt是一个使用TensorRT Python API的PyTorch到TensorRT转换器。 转换器是 易于使用-通过单个函数调用torch2trt转换模块 易于扩展-用Python编写自己的层转换器，并在@tensorrt_converter注册 如果您发现问题，请！ 请注意，此转换器对TensorRT / PyTorch的覆盖范围有限。 我们创建它主要是为了轻松优化项目中使用的模型。 如果您发现转换器对其他型号有用，请。 用法 以下是一些使用示例，有关更多信息，请查看。 转变 import torch from torch2trt import torch2trt from torchvision . models . alexnet import alexnet # create some regular pytorch model... model