Autoencoder on MNIST dataset