冬落 преди 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