torch2trt:易于使用的PyTorch到TensorRT转换器 源码

火炬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