MONAI icon indicating copy to clipboard operation
MONAI copied to clipboard

Support TensorRT export with dynamo

Open binliunls opened this issue 2 years ago • 0 comments

Is your feature request related to a problem? Please describe. PyTorch introduced the dynamo as a part to compile PyTorch models to different backends. And it also works for the Torch-TensorRT compiler and the Torch-TensorRT team is working on supporting it. Therefore, it would be better to have MONAI trt_export API also support this feature.

Describe the solution you'd like Add the use_torchscript parameter to the trt_export API to choose whether to use the torchscript as IR. Because the dynamo supports to directly export models from pytorch models instead of the torchscript models.

binliunls avatar Jun 04 '23 12:06 binliunls