1432199ddd553325de0983f0793f400d5db7f7a8,pretrainedmodels/models/torchvision_models.py,,modify_vggs,#,437

Before Change

        return x
        
    // Modify methods
    setattr(model.__class__, "features", features)
    setattr(model.__class__, "logits", logits)
    setattr(model.__class__, "forward", forward)  
    return model

def vgg11(num_classes=1000, pretrained="imagenet"):

After Change

        return x
        
    // Modify methods
    model.features = types.MethodType(features, model)
    model.logits = types.MethodType(logits, model)
    model.forward = types.MethodType(forward, model)
    return model

def vgg11(num_classes=1000, pretrained="imagenet"):
    VGG 11-layer model (configuration "A")

Instances