TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

Convert torchscript model to tensorrt

Open pengxin233 opened this issue 9 months ago • 1 comments

Can I convert the torchscript model to tensorrt format through torch_tensorrt? Is there any corresponding script that you can give me for reference?

pengxin233 avatar Apr 30 '24 08:04 pengxin233

https://pytorch.org/TensorRT/py_api/ts.html?highlight=convert_method#torch_tensorrt.ts.convert_method_to_trt_engine

narendasan avatar Apr 30 '24 20:04 narendasan