@@ -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