Fix SAE for MNIST.