-
Notifications
You must be signed in to change notification settings - Fork 2.5k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add classifier_dropout
param for CrossEncoder.
#2396
Add classifier_dropout
param for CrossEncoder.
#2396
Conversation
One unrelated test fails: @tomaarsen do you know why? |
classifier_dropout
param for CrossEncoder.classifier_dropout
param for CrossEncoder.
@PhilipMay I suspect that there's been a new release of |
Many thanks @tomaarsen ! 🚀 |
Last release is from 20th Dec. 2023: |
I found the bug and reported it: huggingface/huggingface_hub#1956 For this PR I applied a workaround. |
CI is green - can be merged from my point of view. :-) |
This looks great, thanks! And feel free to share any trained models with me, I'm curious to see what you'll cook up :) |
This PR adds the option to set / change the
classifier_dropout
value. This way it is possible to set the dropout value of the classification head. This can be a very important hyperparameter for classification tasks (and also regression).This is a non breaking change: When the
classifier_dropout
isNone
(the default) then nothing happens.For more background please also see:
classifier_dropout
to classification heads huggingface/transformers#12794classifier_dropout
in model config but not using it huggingface/transformers#12792