Exemplos de resnet18 em Python

Exemplo n.º 1

Arquivo: selector.py Projeto: tomato990/BME595

def resnet18(cuda=True, model_root=None):
    print("Building and initializing resnet-18 parameters")
    from imagenet import resnet
    m = resnet.resnet18(True, model_root)
    if cuda:
        m = m.cuda()
    return m, dataset.get, True

Exemplo n.º 2

Arquivo: selector.py Projeto: YJieZhang/pytorch-classification

def resnet18(cuda=True, model_root=None):
    print("Building and initializing resnet-18 parameters")
    from imagenet import resnet
    m = resnet.resnet18(True, model_root)
    if cuda:
        m = m.cuda()
    return m, dataset.get, True