Returning logits from Trainer.predict()
Hello!I would like to perform some operations on the output probability distributions of anAutoModelForSequenceClassificationmodel so, I was wondering if it is possible to return the logits rather than predicted class labels from thetransformers.Trainer.predict()method.
2021-08-30T18:12:22Z
The predict method does return the logits, as well as the labels.
2021-08-31T04:53:28Z
Thank you for the reply, I realized how stupid of a mistake mine was, but I couldn’t delete the post. It surely does return the logits.
2021-08-31T10:53:07Z
No worries, there are no stupid questions