Skip to content

Instantly share code, notes, and snippets.

@prateekjoshi565

prateekjoshi565/load_weights_bert.py

Created July 18, 2020 11:16
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save prateekjoshi565/87a958e0bc42e175408fbabb49cd8842 to your computer and use it in GitHub Desktop.
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