Function load_ResNet152Model

src/modules/pixie/models/resnet.py:182–185

()

    return model

def load_ResNet152Model():
    model = ResNet(Bottleneck, [3, 8, 36, 3])
    copy_parameter_from_resnet(model, torchvision.models.resnet152(pretrained = True).state_dict())
    return model

# model.load_state_dict(checkpoint['model_state_dict'])