Hey guys,
I am currently using the Trainer 4 in order to train my DistilBertForSequenceClassification.
My problem: I want to stepwise print/save the loss and accuracy of my training set by using the Trainer. Is there a way to do so?
What I did so far: I have adjusted compute_metrics. But this function is only carried out on my evaluation set. I need the same for my training set. I also tried out the TrainerCallback 2. But I can’t access the current predictions of the model by using the predefined callbacks.
Another idea would be to customize the Trainer using a custom train function. But firstly I want to ask you whether there is an easier way to do so.
Thank you in advance!