Пример #1
0
def dataset_selector(data):
    if data == 'mnist':
        return dataset.MNIST_loader(args)
    elif data == 'svhn':
        return dataset.SVHN_loader(img_size=args.img_size)
    elif data == 'cifar10':
        return dataset.Cifar10_loader(args)
Пример #2
0
def dataset_selector(data):
    if data == 'mnist':
        return dataset.MNIST_loader(args)
    elif data == 'cifar10':
        return dataset.Cifar10_loader(args)