-
Notifications
You must be signed in to change notification settings - Fork 7k
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
Fix resnet_fpn_backbone(pretrained=True) #7172
Conversation
test/test_extended_models.py
Outdated
@@ -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)], |
There was a problem hiding this comment.
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
There was a problem hiding this 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 🤞
Reviewed By: vmoens Differential Revision: D43116122 fbshipit-source-id: cdd578682e8b643d9e9e4884b986847c1956f1d3
This fixes
resnet_fpn_backbone(pretrained=True)
which I broken in #7100Thanks Philip for noticing!