val_check_interval equivalent for training loss logging
I very much like the feature of val_check_interval. I use it for logging the validation loss when the epochs are too large. However, I would like to log the training loss at the same steps as well so that for every validation loss entry in my logs, I also have a corresponding training loss entry.
Is there an easy way to do it, or would I have to write my own logger for that?
Thank you.