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

6
examples/movement-pruning/masked_run_glue.py
View File

@@ -24,12 +24,13 @@ import random
import numpy as np
import torch
from torch.utils.data import DataLoader, RandomSampler, SequentialSampler, TensorDataset
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, TensorDataset
from torch.utils.data.distributed import DistributedSampler
from tqdm import tqdm, trange
from emmental import MaskedBertConfig, MaskedBertForSequenceClassification
from transformers import (
WEIGHTS_NAME,
AdamW,
@@ -43,7 +44,6 @@ from transformers import glue_convert_examples_to_features as convert_examples_t
from transformers import glue_output_modes as output_modes
from transformers import glue_processors as processors
from emmental import MaskedBertConfig, MaskedBertForSequenceClassification
try:
from torch.utils.tensorboard import SummaryWriter