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

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/prateekjoshi565/f0bc5fdf47bae4d5dc5065bc24b86c57.js"></script>
Save prateekjoshi565/f0bc5fdf47bae4d5dc5065bc24b86c57 to your computer and use it in GitHub Desktop.

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