search

fanyang587 / MGL

45 stars 6 forks source link

Pre-trained model #2

Open Leno-B opened 3 years ago

Leno-B commented 3 years ago

Can you provide the Baidu Pan link of the initial model, i.e., resnet50_v2.pth ? Thanks!

def resnet50(pretrained=False, **kwargs):
    model = ResNet(Bottleneck, [3, 4, 6, 3], **kwargs)
    if pretrained:
        # model.load_state_dict(model_zoo.load_url(model_urls['resnet50']))
        model_path = './initmodel/resnet50_v2.pth'
        model.load_state_dict(torch.load(model_path), strict=False)
    return model
fido20160817 commented 1 year ago

+1