Accessing logits when using Trainer API
Hi,I’ve been learning to finetune a model using thisguide. Now, for my thesis project I want to access the logits for the training samples at the end of every epoch, so preferably in thecompute-metricsfunction. This function getseval-preds, but is there anyway to have it get access to something liketrain-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!
2024-01-01T18:11:59Z
Hi,This thread may help you:Metrics for Training Set in Trainer - #7 by Kaveri