def __init__(self, task, split='train', transform=None, target_transform=None):
    self.transform = transform
    self.target_transform = target_transform
    self.task = task
    self.split = split
    self.image_roots = (self.task.train_roots if (self.split == 'train') else self.task.test_roots)
    self.labels = (self.task.train_labels if (self.split == 'train') else self.task.test_labels)
def __init__(self, task, split='train', transform=None, target_transform=None):
    self.transform = transform
    self.target_transform = target_transform
    self.task = task
    self.split = split
    self.image_roots = (self.task.train_roots if (self.split == 'train') else self.task.test_roots)
    self.labels = (self.task.train_labels if (self.split == 'train') else self.task.test_labels)