Do I want to apply transformations to augment data inside the CustomDataloader class in Pytorch, or should I do it in the CustomDataset class? I would be grateful if you could show an example of how to use it correctly.