From 66adb71734d27575678e3a67cf1b70d871d0aac1 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Thu, 3 Oct 2019 16:54:40 -0400 Subject: [PATCH] update to transformers --- examples/run_ner.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/run_ner.py b/examples/run_ner.py index 6c6b0f8336..0e40ad02a6 100644 --- a/examples/run_ner.py +++ b/examples/run_ner.py @@ -33,8 +33,8 @@ from torch.utils.data.distributed import DistributedSampler from tqdm import tqdm, trange from utils_ner import convert_examples_to_features, get_labels, read_examples_from_file -from pytorch_transformers import AdamW, WarmupLinearSchedule -from pytorch_transformers import WEIGHTS_NAME, BertConfig, BertForTokenClassification, BertTokenizer +from transformers import AdamW, WarmupLinearSchedule +from transformers import WEIGHTS_NAME, BertConfig, BertForTokenClassification, BertTokenizer logger = logging.getLogger(__name__)