Using torchvision roi_align in libtorch c++ jit modules

C++
1

Hi, I’m trying to use libtorch 1.7.1 to load a jit model that is created with pytorch 1.7.0 and torchvision 0.8.0.
This model is using torchvision::roi_align operator.
When running the model I get this error:

Could not find any similar ops to torchvision::roi_align. This op may not exist or may not be currently supported in TorchScript.

loading the model in pytorch is working fine.
Any idea why its not loading?
I need to install another package to my c++ env to be able to load this model?

Thanks

2

wanted to update that the problem was solved in the attached link, incase someone will encounter the same issue.