Autoencoder Based Data Augmentation Pytorch