You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I have trained a model using @nreimers unsupervised implementation of SimCSE and saved it to the hub.
Now I wish to fine-tune the model on a particular dataset and get predictions.
I follow the implementation specified by Lewis Tunstall in his book -
Use AutoTokenizer to get the tokenizer
Use AutoModelForSequenceClassification to get the model from the hub
Fine-tune it using a Trainer object instance
Question: In the model card for your implementation, you use pooling after getting embeddings using AutoModel.
However, if I have to use the model as a black box using AutoModelForSequenceClassification, can I use it directly, or do I have to manually pool the outputs somewhere as with using AutoModel?
The text was updated successfully, but these errors were encountered:
I have trained a model using @nreimers unsupervised implementation of SimCSE and saved it to the hub.
Now I wish to fine-tune the model on a particular dataset and get predictions.
I follow the implementation specified by Lewis Tunstall in his book -
Question: In the model card for your implementation, you use pooling after getting embeddings using AutoModel.
However, if I have to use the model as a black box using AutoModelForSequenceClassification, can I use it directly, or do I have to manually pool the outputs somewhere as with using AutoModel?
The text was updated successfully, but these errors were encountered: