Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Freeze module pass fails on torchvision RCNN models #45902

Closed
neginraoof opened this issue Oct 6, 2020 · 0 comments
Closed

Freeze module pass fails on torchvision RCNN models #45902

neginraoof opened this issue Oct 6, 2020 · 0 comments
Assignees
@eellison @bzinodev
Labels
oncall: jit Add this issue/PR to JIT oncall triage queue
Projects
JIT Triage

Comments

@neginraoof
Copy link
Contributor

neginraoof commented Oct 6, 2020 •
edited by pytorch-probot bot

馃悰 Bug

Freeze module pass fails with below exception for torchvision Faster RCNN, Mask RCNN, Keypoint RCNN models:

torch._C._freeze_module(script_module._c)
RuntimeError: node->kind() == prim::GetAttr INTERNAL ASSERT FAILED at "../torch/csrc/jit/passes/freeze_module.cpp":22, please report a bug to PyTorch. Expected prim::GetAttr nodes

To Reproduce

Steps to reproduce the behavior:

model = torchvision.models.detection.keypoint_rcnn.keypointrcnn_resnet50_fpn(pretrained=True, min_size=200, max_size=300)
model.eval()
script_module = torch.jit.script(model)
torch._C._freeze_module(script_module._c)

cc @gmagogsfm
@neginraoof neginraoof changed the title Freeze module pass fails on torchvision Faster RCNN/Mask RCNN/Keypoint RCNN models Freeze module pass fails on torchvision RCNN models Oct 6, 2020
@albanD albanD added the oncall: jit Add this issue/PR to JIT oncall triage queue label Oct 6, 2020
@github-actions github-actions bot added this to Need triage in JIT Triage Oct 6, 2020
@eellison eellison mentioned this issue Oct 12, 2020
Closed
@gmagogsfm gmagogsfm moved this from Need triage to Pending in JIT Triage Oct 13, 2020
@eellison eellison mentioned this issue Oct 13, 2020
Closed
JIT Triage automation moved this from Pending to Done Oct 15, 2020
@gchanan gchanan mentioned this issue Oct 15, 2020
Closed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@eellison eellison

@bzinodev bzinodev

Labels
oncall: jit Add this issue/PR to JIT oncall triage queue
Projects
JIT Triage
  
Done
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

[JIT] Revert Freezing shared type PR eellison/pytorch
4 participants
@albanD @eellison @neginraoof @bzinodev