def update_gradients(self, grads):
self.sigma2.gradient = grads[<NUM_LIT:0>]<EOL>self.v.gradient = grads[<NUM_LIT:1>]<EOL>
Pull out the gradients, be careful as the order must match the order
in which the parameters are added