fix to sgd maxepochs in mnist example