Exemple #1
0
 def __init__(self, **kwargs):
     """Construct the Cifar10 class."""
     Dataset.__init__(self, **kwargs)
     CIFAR10.__init__(self,
                      root=self.args.data_path,
                      train=self.train,
                      transform=Compose(self.transforms.__transform__),
                      download=self.args.download)
Exemple #2
0
 def __init__(self,
              root="~/data/cifar10",
              train=True,
              download=True,
              transform=None):
     CIFAR10.__init__(self,
                      root=root,
                      train=train,
                      download=download,
                      transform=transform)