roi_pooling / roi_align ops #2
Description
when I transfer a pytorch model with torchvision.roi_align operation, the inference procedure failed with following Error:
poptorch.poptorch_core.Error: In poptorch/source/LowerToPopart.cpp:913: 'poptorch_cpp_error': Couldn't find a registered operation for node
Error raised in:
[0] processing %roi_feature_map.1 : Float(3, 256, 8, 2, strides=[4096, 16, 2, 1], requires_grad=1, device=cpu) = torchvision::roi_align(%6051, %6106, %4804, %4805, %4806, %4807, %4808) # /usr/local/lib/python3.6/site-packages/torchvision/ops/roi_align.py:57:0
[1] LowerToPopartImpl::lowerBody
how can I solve it?