Accessing logits when using Trainer API

Hi,

I’ve been learning to finetune a model using this guide. Now, for my thesis project I want to access the logits for the training samples at the end of every epoch, so preferably in the compute-metrics function. This function gets eval-preds, but is there anyway to have it get access to something like train-preds? If it’s unclear, I’m looking to log the logits of every training sample at the end of every epoch.

Thanks in advance!

Hi,

This thread may help you: Metrics for Training Set in Trainer - #7 by Kaveri