def setup_class(cls): dataset = add_ids(MNIST)( "./TestSinglePartitionDataset", download=True, transform=transforms.ToTensor(), ) dataset1, dataset2 = partition_dataset(dataset) cls.dataset1 = dataset1 cls.dataset2 = dataset2
def setup_class(cls): cls.dataset = add_ids(MNIST)("./TestPartition", download=True, transform=transforms.ToTensor())
def setup_class(cls): cls.dataset = add_ids(MNIST)( "./TestVerticalDataLoader", download=True, transform=transforms.ToTensor(), )