Autoencoder Based Data Augmentation Pytorch Documentation