Viewing loss on TensorBoard for both the train data set and the validation data set while training a custom object detection model in TensorFlow 2

Question

I have trained a custom object detection model using TensorFlow 2 and the instructions in this colab notebook.

To avoid over- or under-fitting the model, I would like to