Skip to content

roi_pooling / roi_align ops #2

@hzh8311

Description

@hzh8311

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?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions