I used torchvision.transforms.ColorJitter to do data augmentation. The codes are as below:
data_transform = transforms.Compose([
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