Python Data Augmentation Pytorch Vs Sklearn Library