冬落 пре 2 година
родитељ
комит
c553264745
1 измењених фајлова са 1 додато и 1 уклоњено
  1. 1 1
      models/detectors/yolov1/yolov1_backbone.py

+ 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