Make autoencoder test smaller, so that it runs faster