Skip to content

Conversation

@stephantul
Copy link
Contributor

This PR adds a classifier freezing flag that does not update the original embedding weights at all. This allows users to precompute embeddings and load the same model for multiple classification tasks.

Closes #272

Copy link
Member

@Pringled Pringled left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@stephantul stephantul merged commit f0bb56f into main Sep 7, 2025
7 checks passed
@stephantul stephantul deleted the freeze-classifier branch September 7, 2025 09:37
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

Successfully merging this pull request may close these issues.

Feature request: allow users to freeze model weights when training a classifier

3 participants