bugfix: Correctly set learning rate from learning rate finder on restart