Python get_pretrained_model Exemples

Exemple #1

def _wrn(arch, model, pretrained=False, progress=True, dataset="CIFAR10"):
    model = get_pretrained_model(arch,
                                 dataset,
                                 model,
                                 model_urls,
                                 pretrained=pretrained,
                                 progress=progress)
    return model

Exemple #2

Fichier : resnet.py Projet : CYVincent/Scene-Graph-Transformer-CogTree

def _ResNet(arch,
            *args,
            pretrained=False,
            progress=True,
            dataset='CIFAR10',
            **kwargs):
    model = ResNet(*args, **kwargs)
    model = get_pretrained_model(arch,
                                 dataset,
                                 model,
                                 model_urls,
                                 pretrained=pretrained,
                                 progress=progress)
    return model