예제 #1
0
파일: fmnist.py 프로젝트: zhwzhong/vega
 def __init__(self, **kwargs):
     """Construct the Imagenet class."""
     Dataset.__init__(self, **kwargs)
     FashionMNIST.__init__(self,
                           root=self.args.data_path,
                           train=self.train,
                           transform=Compose(self.transforms.__transform__),
                           download=self.args.download)
예제 #2
0
파일: fmnist.py 프로젝트: vineetrao25/vega
 def __init__(self, **kwargs):
     """Construct the Imagenet class."""
     Dataset.__init__(self, **kwargs)
     self.args.data_path = FileOps.download_dataset(self.args.data_path)
     FashionMNIST.__init__(self, root=self.args.data_path, train=self.train,
                           transform=self.transforms, download=self.args.download)