diff --git a/examples/run_tf_glue.py b/examples/run_tf_glue.py index 9235612cb0f564..f2e94ae39e92f1 100644 --- a/examples/run_tf_glue.py +++ b/examples/run_tf_glue.py @@ -1,6 +1,6 @@ import tensorflow as tf import tensorflow_datasets -from transformers import * +from transformers import BertTokenizer, TFBertForSequenceClassification, glue_convert_examples_to_features, BertForSequenceClassification # Load dataset, tokenizer, model from pretrained model/vocabulary tokenizer = BertTokenizer.from_pretrained('bert-base-cased')