Set the learning rate