Torch_tensorrt compile doesn’t support pretrained torchvision Mask_RCNN model.
temporary: the only valid use of a module is looking up an attribute but found = prim::SetAttr[name=“_has_warned”](%self, %178)
TensorRT Version: 1.2.0a0 (torch_tensorrt)
GPU Type: GeForce RTX 3070
Nvidia Driver Version: 470.141.03
CUDA Version: 11.4
Operating System + Version: Ubuntu 20.04
Python Version (if applicable): 3.8.13
TensorFlow Version (if applicable): -
PyTorch Version (if applicable): 1.13.0a0+08820cb
Baremetal or Container (if container which image + tag): Nvidia NGC Container [nvcr.io/nvidia/pytorch:22.07-py3]
Steps To Reproduce
Please execute the .ipynb file provided above.
We recommend you to check the supported features from the below link.
You can refer below link for all the supported operators list.
For unsupported operators, you need to create a custom plugin to support the operation
Thank you for the response. However, I am using torch_tensorrt (via a NGC container) which I don’t think uses onnx.
We recommend you to please reach out to Issues · pytorch/TensorRT · GitHub to get better help.