Normalization is required

Using normalization in custom transforms required as below:
```python

normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406],
                                 std=[0.229, 0.224, 0.225])
```