Skip to content

Instantly share code, notes, and snippets.

@eerkaijun

eerkaijun/load_model.py

Created August 14, 2020 14:05
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 eerkaijun/ad3f354099938c5ac1911ea97711ed0b to your computer and use it in GitHub Desktop.
Save eerkaijun/ad3f354099938c5ac1911ea97711ed0b to your computer and use it in GitHub Desktop.
Download ZIP
Load BERT pretrained model
Raw
load_model.py
# Requirements: install HuggingFace Transformer library beforehand
# pip install transformers
from transformers import BertTokenizer, TFBertForSequenceClassification
from transformers import InputExample, InputFeatures
model = TFBertForSequenceClassification.from_pretrained("bert-base-uncased")
tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment