Exemple #1
0
    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
Exemple #2
0
 def setup_class(cls):
     cls.dataset = add_ids(MNIST)("./TestPartition",
                                  download=True,
                                  transform=transforms.ToTensor())
Exemple #3
0
 def setup_class(cls):
     cls.dataset = add_ids(MNIST)(
         "./TestVerticalDataLoader", download=True, transform=transforms.ToTensor(),
     )