minor fixes to bench_mnist.