def dataset_selector(data): if data == 'mnist': return dataset.MNIST_loader(img_size=args.img_size) elif data == 'svhn': return dataset.SVHN_loader(img_size=32) elif data == "usps": return dataset.usps_loader(img_size=args.img_size) elif data == "mnistm": return dataset.MNIST_M_loader(img_size=args.img_size) elif data == "cifar10": return dataset.cifar10_loader(args)
def dataset_selector(db): if db == "cifar10": return dataset.cifar10_loader(args) else: return dataset.cifar100_loader(args)
def dataset_selector(): # return dataset.cifar100_loader(args) return dataset.cifar10_loader(args)