From da2e47ad15e552b84815da20daf3282b517103f7 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Fri, 27 Sep 2019 09:41:15 +0200 Subject: [PATCH] clean up a little run_tf_glue --- examples/run_tf_glue.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/run_tf_glue.py b/examples/run_tf_glue.py index 9235612cb0..f2e94ae39e 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')