I am fine-tuning for a classification task - I am trying to replicate (and potentially replace) my native PyTorch training and evaluation loops with the Trainer API.
I usually log metrics for both training and validation across each batch/epoch.
Here is what I can achieve with Trainer API.

The accuracy and F1 are of validation sets and I want to also see the same set of metrics at each step for training data as well. Can someone guide on how I can achieve this?
Thanks.