-
Notifications
You must be signed in to change notification settings - Fork 1.1k
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
Do not load pretrained VGG #466
Conversation
Codecov Report
@@ Coverage Diff @@
## master #466 +/- ##
==========================================
- Coverage 80.54% 80.35% -0.20%
==========================================
Files 190 190
Lines 10338 10260 -78
Branches 1533 1533
==========================================
- Hits 8327 8244 -83
- Misses 1781 1783 +2
- Partials 230 233 +3
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
mmedit/models/extractors/lte.py
Outdated
@@ -30,7 +30,7 @@ def __init__(self, requires_grad=True, pixel_range=1., pretrained=None): | |||
pixel_range=pixel_range, img_mean=vgg_mean, img_std=vgg_std) | |||
|
|||
# use vgg19 weights to initialize | |||
vgg_pretrained_features = models.vgg19(pretrained=True).features | |||
vgg_pretrained_features = models.vgg19(pretrained=False).features |
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.
What's the motivation of this change?
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.
Avoid downloading VGG in unittest of CI.
* Don not load pretrained VGG * Add load_pretrained_vgg Co-authored-by: liyinshuo <liyinshuo@sensetime.com>
No description provided.