Fix MNIST benchmark.
Fix CIFAR.