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.

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

Already on GitHub? Sign in to your account

Fix resnet_fpn_backbone(pretrained=True) #7172

Merged

Conversation

NicolasHug
Copy link
Member

This fixes resnet_fpn_backbone(pretrained=True) which I broken in #7100

Thanks Philip for noticing!

@@ -425,7 +426,8 @@ def builder(*, weights=None, flag):
+ TM.list_model_fns(models.quantization)
+ TM.list_model_fns(models.segmentation)
+ TM.list_model_fns(models.video)
+ TM.list_model_fns(models.optical_flow),
+ TM.list_model_fns(models.optical_flow)
+ [lambda pretrained: resnet_fpn_backbone(backbone_name="resnet50", pretrained=pretrained)],
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

can confirm it fails on main

Copy link
Collaborator

@pmeier pmeier left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks Nicolas. Lint workflow is still red, but this time it is the C lint that is acting up. Let's hope this was just a fluke 🤞

@NicolasHug NicolasHug merged commit 2cd25c1 into pytorch:main Feb 6, 2023
@NicolasHug NicolasHug added module: models other if you have no clue or if you will manually handle the PR in the release notes labels Feb 6, 2023
facebook-github-bot pushed a commit that referenced this pull request Feb 9, 2023
Reviewed By: vmoens

Differential Revision: D43116122

fbshipit-source-id: cdd578682e8b643d9e9e4884b986847c1956f1d3
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cla signed module: models other if you have no clue or if you will manually handle the PR in the release notes
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants