repair MomentumGradientDescent