Closed dblakely closed 2 months ago
"id2label": {
"0": "LABEL_0"
},
means you only have 1 class -> the above config.json is identical to rerankers.
Loads reranker. Needs at least two trained classes to work with infinity.
Thanks for the quick reply. So the classification layer is Linear(in_features=1536, out_features=1, bias=True)
. Is it fine to run this model as a reranker then to get scores? If so, what do you set the query to?
Well, your model is a two class classifier, it bases multi_label_classification
but has once classification head dim = 1. My suggestion is is to not create single classification head dim, but use dim=2 for your problem.
System Info
Running
infinity
via docker (michaelf34/infinity:latest
) + using the REST API to call the modelInformation
Tasks
Reproduction
Which gives me:
The model's config looks like:
So it's specified that the model is for classification in the config.
Expected behavior
While the base deberta-v2 doesn't have a classification head, this checkpoint does and I finetuned it for classification. So I was expecting the
/classify
endpoint to work in this case. Is there anything wrong with the config or is there a way to have Infinity run a deberta-v2 finetuned for sequence classification?