search

graphcore / poptorch

PyTorch interface for the IPU
https://docs.graphcore.ai/projects/poptorch-user-guide/en/latest/
MIT License
176 stars 14 forks source link

roi_pooling / roi_align ops #2

Open hzh8311 opened 2 years ago

hzh8311 commented 2 years ago

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?