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
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