Python ConcatDataset.__init__ Examples

Example #1

File: task.py Project: zzf2014/pytorch-meta

 def __init__(self, datasets, num_classes, target_transform=None):
     Task.__init__(self, num_classes)
     ConcatDataset.__init__(self, datasets)
     for task in self.datasets:
         task.target_transform_append(target_transform)

Example #2

 def __init__(self, datasets):
     Dataset.__init__(self)
     TorchConcatDataset.__init__(self, datasets)

Example #3

 def __init__(self, datasets: Iterable[Distributed], tag=None):
     datasets = list(datasets)
     statTitle = unique(sum([i.statTitle for i in datasets], []))
     self.tag = tag
     ConcatDataset.__init__(self, datasets)
     Distributed.__init__(self, statTitle)

Example #4

 def __init__(self, datasets: Sequence[MultifieldDataset]):
     ConcatDataset.__init__(self, datasets=datasets)

Example #5

File: task.py Project: struemya/INR

 def __init__(self, datasets, num_classes, target_transform=None):
     index = tuple(task.index for task in datasets)
     Task.__init__(self, index, num_classes)
     ConcatDataset.__init__(self, datasets)
     for task in self.datasets:
         task.target_transform_append(target_transform)