On the convergence of the Stochastic Heavy Ball Method