From 9fc8dcb2a071b0b00d05d75b4a528f2dee4b2d00 Mon Sep 17 00:00:00 2001 From: Aymeric Augustin Date: Sun, 22 Dec 2019 20:17:04 +0100 Subject: [PATCH] Standardize import. Every other file uses this pattern. --- examples/run_ner.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/examples/run_ner.py b/examples/run_ner.py index 63f81b077e..d7972c3415 100644 --- a/examples/run_ner.py +++ b/examples/run_ner.py @@ -25,7 +25,6 @@ import random import numpy as np import torch from seqeval.metrics import f1_score, precision_score, recall_score -from tensorboardX import SummaryWriter from torch.nn import CrossEntropyLoss from torch.utils.data import DataLoader, RandomSampler, SequentialSampler, TensorDataset from torch.utils.data.distributed import DistributedSampler @@ -54,6 +53,12 @@ from transformers import ( from utils_ner import convert_examples_to_features, get_labels, read_examples_from_file +try: + from torch.utils.tensorboard import SummaryWriter +except ImportError: + from tensorboardX import SummaryWriter + + logger = logging.getLogger(__name__) ALL_MODELS = sum(