On adaptive stochastic heavy ball momentum for solving linear systems