the reason for add_classification_head

[rabeehkarimimahabadi]

the reason for add_classification_head

[calpt]

The add_classification_head() method is our implementation of adding a final task-specific prediction head on top of the last layer of the Transformer model, in this case a simple FF layer to map the final hidden state to one of n classes in a sequence classification problem. This is specific to our AutoModelWithHeads class to allow flexibly adding and switching between multiple prediction heads. Therefore, this is not needed in case you don't need this kind of final prediction head layer or provide your own.

[rabeehkarimimahabadi]

thank you