Exemplos de cifar100 em Python

Exemplo n.º 1

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

def cifar100(cuda=True, model_root=None):
    print("Building and initializing cifar100 parameters")
    from cifar import model, dataset
    m = model.cifar100(128, pretrained=os.path.join(model_root, 'cifar100.pth'))
    if cuda:
        m = m.cuda()
    return m, dataset.get100, False

Exemplo n.º 2

Arquivo: selector.py Projeto: wjc852456/pytorch-quant

def cifar100(cuda=True, model_root=None):
    print("Building and initializing cifar100 parameters")
    from cifar import model, dataset
    m = model.cifar100(128, pretrained=os.path.join(model_root, 'cifar100.pth'))
    if cuda:
        m = m.cuda()
    return m, dataset.get100, False

Exemplo n.º 3

Arquivo: selector.py Projeto: luk1684tw/Precision-Reduction

def cifar100(cuda=True, model_root=None, **kwargs):
    print("Building and initializing cifar100 parameters")
    from cifar import model, dataset

    use_model_zoo = False if model_root else True
    m = model.cifar100(pretrained=os.path.join(model_root, 'cifar100.pth'),
                       use_model_zoo=use_model_zoo,
                       **kwargs)
    if cuda:
        m = m.cuda()
    return m, dataset.get100, False