Skip to content
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

Model.fit not saving the config.json file correctly for Crossencoders #2649

Open
edacul opened this issue May 15, 2024 · 1 comment
Open

Comments

@edacul
Copy link

edacul commented May 15, 2024

Hi, I have a multi label classificator train mounted on a sentence transformers cross encoder that outputs multi class labels for encoded text BUT somehow the training code seems to have changed and now the config.json doesn't retains the correct labels2id and id2label properties:

image

Edit: the labels are defined this way:

image

This is used to work just fine

Thank you for your help!

@tomaarsen
Copy link
Collaborator

Hello!

What (new) Sentence Transformers version are you using, and do you know which version did work well?

  • Tom Aarsen

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants