Hey guys,
I used torchvision.transforms.ColorJitter to do data augmentation. The codes are as below:
data_transform = transforms.Compose([
transforms.ColorJitter(brightness=0.1,constrast=0.1,saturation=0.1,hue=0.1),
transforms.ToTensor(),
transforms.Normalize([pixel_mean, pixel_mean, pixel_mean], [pixel_std, pixel_std, pixel_std])
])
But it gave the error of 'AttributeError: ‘module’ object has no attribute ‘ColorJitter’'
My Pytorch version is 0.2.0_4.  Any help? Thanks in advance