You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
The text was updated successfully, but these errors were encountered:
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 thetrt_export
API to choose whether to use the torchscript as IR. Because thedynamo
supports to directly export models from pytorch models instead of the torchscript models.The text was updated successfully, but these errors were encountered: