-
Notifications
You must be signed in to change notification settings - Fork 364
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
How to freeze params of an embedding table in EmbeddingBagCollection? #1990
Comments
Hi, @henrylhtsang @IvanKobzarev @joshuadeng @PaulZhang12 can you see this problem? |
a few ways you could do this: Keep in mind htat if you have only some tables you want to freeze (and some you want to always be trainable) that you wil likely want to create separate EBCs for each (otherwise the sharder might combine them) |
Thanks! |
No description provided.
The text was updated successfully, but these errors were encountered: