Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

[JIT] Not supported for maskrcnn_resnet50_fpn

See original GitHub issue

I am trying to accelerate the maskrcnn_resnet50_fpn pretrained model using JIT tracing provided by pytorch. It appears that some operations present in this model are not supported by pytorch JIT.

Are these models supposed to have JIT support officially? If not, would you be able to provide advice for a workaround?

To replicate, running:

import torch
import torchvision
model = torchvision.models.detection.maskrcnn_resnet50_fpn(pretrained=True)
model.eval()
traced_net = torch.jit.trace(model, torch.rand(1, 3,800, 800))

produces

RuntimeError: log2_vml_cpu not implemented for 'Long

Thank you.

Issue Analytics

State:
Created 4 years ago
Reactions:7
Comments:59 (27 by maintainers)

Top GitHub Comments

6reactions

eellisoncommented, Aug 29, 2019

@cted18 Yes i’ll be working on adding OrderedDict to support fcn_resnet101. I think together with op support added in https://github.com/pytorch/vision/pull/1267 it shouldn’t be too hard to support in script.

4reactions

fmassacommented, Dec 2, 2019

@stereomatchingkiss ONNX and JIT support for Mask R-CNN in torchvision has been merged into master, and is available if you compile from source.

Top Results From Across the Web

Torch.jit.trace is not working with mask rcnn

maskrcnn_resnet50_fpn ? torch.jit.script and torch.jit.tarce are not working with this model. With torch.jit.script model = torch ...

[PyTorch] Failed to convert Mask R-CNN with batch size > 1

Recerntly I am working on some models from PyTorch implementation, so I use the Relay PyTorch frontend from_pytorch to convert the model.

I can't convert the pytorch maskrcnn_resnet50_fpn model ...

But The problem is not in some dependencies. The problen is that *.pt model doesn't load with module = torch::jit::load(_path); The program ...

PyTorch for Jetson - Jetson Nano - ...

_C import * ImportError: libnvToolsExt.so.1: cannot open shared object file: No such file or ... Yes I did the same thing to get...

hpa_fastrcnn

__init__() if not isinstance(min_size, (list, tuple)): min_size = (min_size,) ... it in-place # once torchscript supports dict comprehension # this can be ...