冬落 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