Autoencoder Pytorch Example Project