Torch_tensorrt compile doesn't support pretrained torchvision Mask_RCNN model


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
CUDNN Version:
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 []

Relevant Files

Steps To Reproduce

Please execute the .ipynb file provided above.

Hi ,
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.

Thank you.