Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "869093e890ffface86430e378cc71ebcc080f2cf"
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
Hide 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, * ...@@ -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') raise NotImplementedError('This model does not use auxiliary loss')
backbone_name = 'mobilenet_v3_large' backbone_name = 'mobilenet_v3_large'
if pretrained:
kwargs["pretrained_backbone"] = False
model = _segm_lraspp_mobilenetv3(backbone_name, num_classes, **kwargs) model = _segm_lraspp_mobilenetv3(backbone_name, num_classes, **kwargs)
if pretrained: 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