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
Posted this on Stack overflow and 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.