prateekjoshi565/load_weights_bert.py

Created July 18, 2020 11:16

Star () You must be signed in to star a gist
Fork () 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/87a958e0bc42e175408fbabb49cd8842.js"></script>
Save prateekjoshi565/87a958e0bc42e175408fbabb49cd8842 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

load_weights_bert.py

	#load weights of best model
	path = 'saved_weights.pt'
	model.load_state_dict(torch.load(path))

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment