fixing autoencoder demo