Skip to content

Instantly share code, notes, and snippets.

@prateekjoshi565

prateekjoshi565/freeze_model_bert.py

Created July 17, 2020 17:14
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
Embed
What would you like to do?
Learn more about clone URLs
Download ZIP
Raw
freeze_model_bert.py
# freeze all the parameters
for param in bert.parameters():
param.requires_grad = False
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment