Skip to content

Instantly share code, notes, and snippets.

@prateekjoshi565
Created July 17, 2020 17:28
  • 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
Save prateekjoshi565/e41850d47c00803f66b8c1dfe94f21f0 to your computer and use it in GitHub Desktop.
# pass the pre-trained BERT to our define architecture
model = BERT_Arch(bert)
# push the model to GPU
model = model.to(device)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment