Skip to content

GitLab

Commit 66adb717 authored by thomwolf's avatar thomwolf
Browse files

update to transformers

parent 5ff9cd15
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/run_ner.py
View file @ 66adb717
...@@ -33,8 +33,8 @@ from torch.utils.data.distributed import DistributedSampler ...@@ -33,8 +33,8 @@ from torch.utils.data.distributed import DistributedSampler
from tqdm import tqdm, trange from tqdm import tqdm, trange
from utils_ner import convert_examples_to_features, get_labels, read_examples_from_file from utils_ner import convert_examples_to_features, get_labels, read_examples_from_file
from pytorch_transformers import AdamW, WarmupLinearSchedule from transformers import AdamW, WarmupLinearSchedule
from pytorch_transformers import WEIGHTS_NAME, BertConfig, BertForTokenClassification, BertTokenizer from transformers import WEIGHTS_NAME, BertConfig, BertForTokenClassification, BertTokenizer
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment