Python Data Augmentation Pytorch Vs Sklearn Pipeline