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

11
examples/movement-pruning/emmental/modeling_bert_masked.py
View File

@@ -26,13 +26,16 @@ import torch
from torch import nn
from torch.nn import CrossEntropyLoss, MSELoss
from emmental import MaskedBertConfig, MaskedLinear
from transformers.file_utils import add_start_docstrings, add_start_docstrings_to_callable
from transformers.modeling_bert import (
ACT2FN,
BERT_PRETRAINED_MODEL_ARCHIVE_MAP,
BertLayerNorm,
load_tf_weights_in_bert,
)
from transformers.modeling_utils import PreTrainedModel, prune_linear_layer
from transformers.modeling_bert import BERT_PRETRAINED_MODEL_ARCHIVE_MAP
from transformers.modeling_bert import load_tf_weights_in_bert, ACT2FN, BertLayerNorm
from emmental import MaskedLinear
from emmental import MaskedBertConfig
logger = logging.getLogger(__name__)