https://medium.com/the-artificial-impostor/pytorch-lightning-log-training-losses-when-accumulating-gradients-d33abfca2874