def get_dataset(name): if name == "MNIST": return data.MNIST() elif name == "CIFAR100": return data.CIFAR100() elif name == "CIFAR10": return data.CIFAR10() else: print("Unsupported Dataset") assert False
def get_dataset(name): if name == 'CIFAR10': return data.CIFAR10() elif name == "CIFAR100": return data.CIFAR100() elif name == "Imagenet": return data.Imagenet() elif name == "VggFace2_1K": return data.VggFace2_1K() elif name == "VggFace2_5K": return data.VggFace2_5K() elif name == "Google_Landmark_v2_1K": return data.Google_Landmark_v2_1K() elif name == "Google_Landmark_v2_10K": return data.Google_Landmark_v2_10K()
def get_dataset(name): if name == 'CIFAR100': return data.CIFAR100() elif name == "TinyImagenet": return data.TinyImagenet()
def get_dataset(name): if name == "CIFAR100": return data.CIFAR100() elif name == 'MNIST': return data.MNIST()