Skip to content

GitLab

Unverified Commit ddd14ffc authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Don't download backbone weights if pretrained=True (#4283)

parent 8e2bd0e0
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
torchvision/models/segmentation/segmentation.py
View file @ ddd14ffc
......@@ -186,6 +186,8 @@ def lraspp_mobilenet_v3_large(pretrained=False, progress=True, num_classes=21, *
raise NotImplementedError('This model does not use auxiliary loss')
backbone_name = 'mobilenet_v3_large'
if pretrained:
kwargs["pretrained_backbone"] = False
model = _segm_lraspp_mobilenetv3(backbone_name, num_classes, **kwargs)
if pretrained:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment