8961e3d5f50c689c5a406a88768b31de0059f8b4,gluoncv/model_zoo/resnetv1b.py,,resnet34_v1b,#,287

Before Change

    model = ResNetV1b(BasicBlockV1b, [3, 4, 6, 3], **kwargs)
    if pretrained:
        from .model_store import get_model_file
        model.load_params(get_model_file("resnet%d_v%db"%(34, 1),
                                         tag=pretrained, root=root), ctx=ctx)
    return model


def resnet50_v1b(pretrained=False, root="~/.mxnet/models", ctx=cpu(0), **kwargs):

After Change

    model = ResNetV1b(BasicBlockV1b, [3, 4, 6, 3], **kwargs)
    if pretrained:
        from .model_store import get_model_file
        model.load_params(get_model_file("resnet%d_v%db"%(34, 1),
                                         tag=pretrained, root=root), ctx=ctx)
        from ..data import ImageNet1kAttr
        attrib = ImageNet1kAttr()
        model.synset = attrib.synset
        model.classes = attrib.classes

Instances