SGD with momentum in Keras