Understanding Token classification head
I was trying to understand how AutoModels work in hugging face. There is a proper description for BertForTokenClassification’s classification heads but not for the generic AutoModelForTokenClassificaton. I wanna know what is this this token classification head so that I can use it directly or write a custom model with a required classification head
2023-04-04T12:14:58Z
Posted this onStack overflowand it’s just a Dropout layer and a linear layer attached to it. Refer to the link for more details on how to check this through code.