-
Notifications
You must be signed in to change notification settings - Fork 211
Conversation
Codecov Report
@@ Coverage Diff @@
## master #237 +/- ##
==========================================
+ Coverage 86.90% 86.95% +0.05%
==========================================
Files 58 58
Lines 3055 3067 +12
==========================================
+ Hits 2655 2667 +12
Misses 400 400
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
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 !
flash/vision/backbones.py
Outdated
result = fn(*args, pretrained=False, **kwargs) | ||
rank_zero_warn( | ||
"Failed to download pretrained weights for the selected backbone. The backbone has been created with" | ||
" `pretrained=False` instead. If you are loading from a local checkpoint, this warning can be safely" | ||
" ignored.", UserWarning | ||
) |
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.
Changed the order so if the fn call fails too, the warning is still shown
result = fn(*args, pretrained=False, **kwargs) | |
rank_zero_warn( | |
"Failed to download pretrained weights for the selected backbone. The backbone has been created with" | |
" `pretrained=False` instead. If you are loading from a local checkpoint, this warning can be safely" | |
" ignored.", UserWarning | |
) | |
rank_zero_warn( | |
"Failed to download pretrained weights for the selected backbone. The backbone has been created with" | |
" `pretrained=False` instead. If you are loading from a local checkpoint, this warning can be safely" | |
" ignored.", UserWarning | |
) | |
result = fn(*args, pretrained=False, **kwargs) |
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.
Not sure we want this. If it fails with pretrained=False
then the warning is no longer true when it says "the backbone has been created"?
What does this PR do?
Fixes #155
Before submitting
PR review
Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in Github issues there's a high chance it will not be merged.
Did you have fun?
Make sure you had fun coding 🙃