Fix learning rate stopping criteria