冬落 2 rokov pred
rodič
commit
c553264745

+ 1 - 1
models/detectors/yolov1/yolov1_backbone.py

@@ -217,7 +217,7 @@ def resnet152(pretrained=False, **kwargs):
     """
     model = ResNet(Bottleneck, [3, 8, 36, 3], **kwargs)
     if pretrained:
-        model.load_state_dict(model_zoo.load_url(model_urls['resnet152']))
+        model.load_state_dict(model_zoo.load_url(model_urls['resnet152']), strict=False)
     return model
 
 ## build resnet