79658049

Date: 2025-06-08 17:41:20
Score: 1.5
Natty:
Report link

AutoModelForSequenceClassification creates a linear mapping nn.Linear(config.n_embd, config.num_labels) and allows you to change num_labels via config.

So for example you can do AutoModelForSequenceClassification.from_pretrained(MODEL, num_labels=3)

But also if you do that, you need to train the model otherwise the outputs are going to be random.

Reasons:
  • No code block (0.5):
  • Low reputation (1):
Posted by: Baki