Python Data Augmentation Pytorch Dataloader