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

Make text encoder trainable property default to False for pre-trained HF encoders #2060

Merged

Conversation

dantreiman
Copy link
Collaborator

Our documentation says that the default value of trainable is False. Updating code to match that:
https://ludwig-ai.github.io/ludwig-docs/0.5/configuration/features/text_features/#huggingface-models

This also matches expectations learned from user feedback: if using a pretrained model, default behavior should be to freeze the pretrained model and train the prediction layers using the pretrained model's embeddings.
\

Copy link
Collaborator

@justinxzhao justinxzhao left a comment

Choose a reason for hiding this comment

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

Thanks!

@github-actions
Copy link

Unit Test Results

       6 files         6 suites   2h 21m 5s ⏱️
2 798 tests 2 763 ✔️   35 💤 0
8 394 runs  8 285 ✔️ 109 💤 0

Results for commit e39d6d1.

@dantreiman dantreiman merged commit e0a786d into ludwig-ai:master May 26, 2022
@dantreiman dantreiman deleted the daniel/trainable_default_false branch May 26, 2022 15:40
justinxzhao added a commit that referenced this pull request Jun 30, 2022
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.

None yet

2 participants