HI All,
I’m quite new on PyTorch and I have already a interesting challenge ahead. I’m trying to run MaskRCNN (torchvision implementation) on NVIDIA TensorRT SDK. I’ve already reported an issue with them and the initial feedback is that TensorRT doesn’t accept weights exported as tensors.
Is there a way to export  the model (https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html) with python types instead?
Is there any tool I can post run in order to convert the weights?
I really appreciate any feedback.
Thanks a lot