Update train-a-digit-classifier/train-on-mnist.lua