Unverified Commit 36737160 authored by vfdev's avatar vfdev Committed by GitHub
Browse files

Fixes #3322 (#3325)



Disable pretrained backbone downloading if pretrained is True
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent 6116812d
...@@ -69,6 +69,7 @@ def _segm_model(name, backbone_name, num_classes, aux, pretrained_backbone=True) ...@@ -69,6 +69,7 @@ def _segm_model(name, backbone_name, num_classes, aux, pretrained_backbone=True)
def _load_model(arch_type, backbone, pretrained, progress, num_classes, aux_loss, **kwargs): def _load_model(arch_type, backbone, pretrained, progress, num_classes, aux_loss, **kwargs):
if pretrained: if pretrained:
aux_loss = True aux_loss = True
kwargs["pretrained_backbone"] = False
model = _segm_model(arch_type, backbone, num_classes, aux_loss, **kwargs) model = _segm_model(arch_type, backbone, num_classes, aux_loss, **kwargs)
if pretrained: if pretrained:
_load_weights(model, arch_type, backbone, progress) _load_weights(model, arch_type, backbone, progress)
......
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