TensorRT
TensorRT copied to clipboard
Convert torchscript model to tensorrt
Can I convert the torchscript model to tensorrt format through torch_tensorrt? Is there any corresponding script that you can give me for reference?
https://pytorch.org/TensorRT/py_api/ts.html?highlight=convert_method#torch_tensorrt.ts.convert_method_to_trt_engine