Exemplo n.º 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)
Exemplo n.º 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)