Python Code For Gradient Descent With Momentum