(arch, block, layers, pretrained, progress, **kwargs)
| 50 | |
| 51 | |
| 52 | def _resnet(arch, block, layers, pretrained, progress, **kwargs): |
| 53 | model = ResNet(block, layers, **kwargs) |
| 54 | if pretrained: |
| 55 | model_dict = model.state_dict() |
| 56 | pretrained_dict = load_state_dict_from_url(model_urls[arch], |
| 57 | progress=progress) |
| 58 | # remove keys from pretrained dict that doesn't appear in model dict |
| 59 | pretrained_dict = {k: v for k, v in pretrained_dict.items() if k in model_dict} |
| 60 | model.load_state_dict(pretrained_dict, strict=False) |
| 61 | return model |
| 62 | |
| 63 | |
| 64 | def resnet18(pretrained=False, progress=True, **kwargs): |
no test coverage detected