-
Notifications
You must be signed in to change notification settings - Fork 1.4k
Open
Labels
Description
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.
Reactions are currently unavailable