Here is my image transforms code:

train_transform=transforms.Compose([
		transforms.CenterCrop(224),
		transforms.RandomRotation(5),
		transforms.ToTensor(),
#for transfer learning 
		transforms.Normalize(mean = [0.485, 0.456, 0.406],std = [0.229, 0.224, 0.225])])
test_transform=transforms.Compose([
		transforms.RandomCrop(224),
		transforms.RandomRotation(3),
		transforms.ToTensor(),
		transforms.Normalize(mean = [0.485, 0.456, 0.406],std = [0.229, 0.224, 0.225])])