コード例 #1
0
ファイル: task.py プロジェクト: 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)
コード例 #2
0
 def __init__(self, datasets):
     Dataset.__init__(self)
     TorchConcatDataset.__init__(self, datasets)
コード例 #3
0
 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)
コード例 #4
0
 def __init__(self, datasets: Sequence[MultifieldDataset]):
     ConcatDataset.__init__(self, datasets=datasets)
コード例 #5
0
ファイル: task.py プロジェクト: 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)