Skip to content

Instantly share code, notes, and snippets.

@prateekjoshi565

prateekjoshi565/import_bert.py

Created July 17, 2020 13:38
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/232b03bf6471c81537b132abf2f9a92e to your computer and use it in GitHub Desktop.
Save prateekjoshi565/232b03bf6471c81537b132abf2f9a92e to your computer and use it in GitHub Desktop.
Download ZIP
Raw
import_bert.py
# import BERT-base pretrained model
bert = AutoModel.from_pretrained('bert-base-uncased')
# Load the BERT tokenizer
tokenizer = BertTokenizerFast.from_pretrained('bert-base-uncased')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment