"web/git@developer.sourcefind.cn:chenpangpang/ComfyUI.git" did not exist on "8cfce083c4eb09ea95bce59f65f1634e09d12b13"
Commit 9fc8dcb2 authored by Aymeric Augustin's avatar Aymeric Augustin
Browse files

Standardize import.

Every other file uses this pattern.
parent f2522869
...@@ -25,7 +25,6 @@ import random ...@@ -25,7 +25,6 @@ import random
import numpy as np import numpy as np
import torch import torch
from seqeval.metrics import f1_score, precision_score, recall_score from seqeval.metrics import f1_score, precision_score, recall_score
from tensorboardX import SummaryWriter
from torch.nn import CrossEntropyLoss from torch.nn import CrossEntropyLoss
from torch.utils.data import DataLoader, RandomSampler, SequentialSampler, TensorDataset from torch.utils.data import DataLoader, RandomSampler, SequentialSampler, TensorDataset
from torch.utils.data.distributed import DistributedSampler from torch.utils.data.distributed import DistributedSampler
...@@ -54,6 +53,12 @@ from transformers import ( ...@@ -54,6 +53,12 @@ from transformers import (
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
try:
from torch.utils.tensorboard import SummaryWriter
except ImportError:
from tensorboardX import SummaryWriter
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
ALL_MODELS = sum( ALL_MODELS = sum(
......
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