Fix definition of AdaGrad?