more work on stoch_grad_mom.m, need to test