HuggingFace transformers BERT for classification: dimensionality of output with classification layer is expected to be 1, but is 512 instead

Hi, did u managed why the shape change?
I’m having a pretty similar error and cant find nothing usefoul.