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