def __init__(self, train=True, transform=Compose([ToFloat(), Normalize(mean=0.5, std=0.5)]), target_transform=None, label_type='fine'): assert label_type in ['fine', 'coarse'] self.label_type = label_type super().__init__(train, transform, target_transform)
def __init__(self, train=True, transform=Compose([Flatten(), ToFloat(), Normalize(0., 255.)]), target_transform=None): super().__init__(train, transform, target_transform)
def __init__( self, train=True, transform=Compose([ToFloat(), Normalize(mean=0.5, std=0.5)]), target_transform=None, ): super().__init__(train, transform, target_transform)