SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

commplying with isort

Browse Source
This commit is contained in:
Victor SANH
2020-05-28 00:26:39 -04:00
parent db2a3b2e01
commit 5c8e5b3709
9 changed files with 29 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
examples/movement-pruning/masked_run_squad.py
View File

@@ -25,12 +25,13 @@ import timeit
import numpy as np
import torch
from torch.utils.data import DataLoader, RandomSampler, SequentialSampler
from torch.utils.data.distributed import DistributedSampler
import torch.nn as nn
import torch.nn.functional as F
from torch.utils.data import DataLoader, RandomSampler, SequentialSampler
from torch.utils.data.distributed import DistributedSampler
from tqdm import tqdm, trange
from emmental import MaskedBertConfig, MaskedBertForQuestionAnswering
from transformers import (
WEIGHTS_NAME,
AdamW,
@@ -48,8 +49,6 @@ from transformers.data.metrics.squad_metrics import (
from transformers.data.processors.squad import SquadResult, SquadV1Processor, SquadV2Processor
from emmental import MaskedBertConfig, MaskedBertForQuestionAnswering
try:
from torch.utils.tensorboard import SummaryWriter
except ImportError: