diff --git a/examples/contrib/run_swag.py b/examples/contrib/run_swag.py index 58aec25877..8494c5fad9 100644 --- a/examples/contrib/run_swag.py +++ b/examples/contrib/run_swag.py @@ -31,9 +31,13 @@ import torch from torch.utils.data import (DataLoader, RandomSampler, SequentialSampler, TensorDataset) from torch.utils.data.distributed import DistributedSampler -from tqdm import tqdm, trange -from tensorboardX import SummaryWriter +try: + from torch.utils.tensorboard import SummaryWriter +except: + from tensorboardX import SummaryWriter + +from tqdm import tqdm, trange from transformers import (WEIGHTS_NAME, BertConfig, BertForMultipleChoice, BertTokenizer) diff --git a/examples/distillation/distiller.py b/examples/distillation/distiller.py index f736936449..d51bdae77f 100644 --- a/examples/distillation/distiller.py +++ b/examples/distillation/distiller.py @@ -19,7 +19,6 @@ import os import math import psutil import time -from tensorboardX import SummaryWriter from tqdm import trange, tqdm import numpy as np import psutil @@ -31,6 +30,11 @@ from torch.optim import AdamW from torch.utils.data.distributed import DistributedSampler from torch.utils.data import RandomSampler, BatchSampler, DataLoader +try: + from torch.utils.tensorboard import SummaryWriter +except: + from tensorboardX import SummaryWriter + from transformers import WarmupLinearSchedule from utils import logger diff --git a/examples/distillation/run_squad_w_distillation.py b/examples/distillation/run_squad_w_distillation.py index 4be641dd81..a5194d0804 100644 --- a/examples/distillation/run_squad_w_distillation.py +++ b/examples/distillation/run_squad_w_distillation.py @@ -30,9 +30,13 @@ from torch.utils.data import (DataLoader, RandomSampler, SequentialSampler, from torch.utils.data.distributed import DistributedSampler import torch.nn.functional as F import torch.nn as nn -from tqdm import tqdm, trange -from tensorboardX import SummaryWriter +try: + from torch.utils.tensorboard import SummaryWriter +except: + from tensorboardX import SummaryWriter + +from tqdm import tqdm, trange from transformers import (WEIGHTS_NAME, BertConfig, BertForQuestionAnswering, BertTokenizer, diff --git a/examples/run_glue.py b/examples/run_glue.py index e02e9b4294..72f474e012 100644 --- a/examples/run_glue.py +++ b/examples/run_glue.py @@ -28,7 +28,12 @@ import torch from torch.utils.data import (DataLoader, RandomSampler, SequentialSampler, TensorDataset) from torch.utils.data.distributed import DistributedSampler -from tensorboardX import SummaryWriter + +try: + from torch.utils.tensorboard import SummaryWriter +except: + from tensorboardX import SummaryWriter + from tqdm import tqdm, trange from transformers import (WEIGHTS_NAME, BertConfig, diff --git a/examples/run_lm_finetuning.py b/examples/run_lm_finetuning.py index a91deebb6c..442d0e7f6d 100644 --- a/examples/run_lm_finetuning.py +++ b/examples/run_lm_finetuning.py @@ -32,7 +32,12 @@ import numpy as np import torch from torch.utils.data import DataLoader, Dataset, SequentialSampler, RandomSampler from torch.utils.data.distributed import DistributedSampler -from tensorboardX import SummaryWriter + +try: + from torch.utils.tensorboard import SummaryWriter +except: + from tensorboardX import SummaryWriter + from tqdm import tqdm, trange from transformers import (WEIGHTS_NAME, AdamW, WarmupLinearSchedule, diff --git a/examples/run_multiple_choice.py b/examples/run_multiple_choice.py index 54f3a8a904..bea57a9ddc 100644 --- a/examples/run_multiple_choice.py +++ b/examples/run_multiple_choice.py @@ -29,7 +29,12 @@ import torch from torch.utils.data import (DataLoader, RandomSampler, SequentialSampler, TensorDataset) from torch.utils.data.distributed import DistributedSampler -from tensorboardX import SummaryWriter + +try: + from torch.utils.tensorboard import SummaryWriter +except: + from tensorboardX import SummaryWriter + from tqdm import tqdm, trange from transformers import (WEIGHTS_NAME, BertConfig, diff --git a/examples/run_squad.py b/examples/run_squad.py index eb351b340c..43b65d2c3c 100644 --- a/examples/run_squad.py +++ b/examples/run_squad.py @@ -28,9 +28,13 @@ import torch from torch.utils.data import (DataLoader, RandomSampler, SequentialSampler, TensorDataset) from torch.utils.data.distributed import DistributedSampler -from tqdm import tqdm, trange -from tensorboardX import SummaryWriter +try: + from torch.utils.tensorboard import SummaryWriter +except: + from tensorboardX import SummaryWriter + +from tqdm import tqdm, trange from transformers import (WEIGHTS_NAME, BertConfig, BertForQuestionAnswering, BertTokenizer,