How can I check validation loss within an epoch in tensorflow callback?

Question

I am running Tensorflow 2.0 and training a model on 2 GPUs. It takes just over 3,000 batches per epoch and each epoch takes just under an hour. After each epoch, I can see t