I was wondering whether it would be convenient to print additional information such as the learning rate at each iteration of the training. This could be useful for understanding how the training is progressing, IMHO.