Updated autoencoder test to run faster