(arch, block, layers, pretrained, progress, **kwargs)
| 34 | |
| 35 | |
| 36 | def _reid_resnet(arch, block, layers, pretrained, progress, **kwargs): |
| 37 | model = ReidResNet(block, layers, **kwargs) |
| 38 | if pretrained: |
| 39 | model_dict = model.state_dict() |
| 40 | pretrained_dict = load_state_dict_from_url(model_urls[arch], |
| 41 | progress=progress) |
| 42 | # remove keys from pretrained dict that doesn't appear in model dict |
| 43 | pretrained_dict = {k: v for k, v in pretrained_dict.items() if k in model_dict} |
| 44 | model.load_state_dict(pretrained_dict, strict=False) |
| 45 | return model |
| 46 | |
| 47 | |
| 48 | def reid_resnet18(pretrained=False, progress=True, **kwargs): |
no test coverage detected