transformers/examples/movement-pruning/emmental/__init__.py
2020-06-01 09:23:31 -04:00

12 lines
288 B
Python

from .modules import *
from .configuration_bert_masked import MaskedBertConfig
from .modeling_bert_masked import (
MaskedBertModel,
MaskedBertForQuestionAnswering,
MaskedBertForSequenceClassification,
MaskedBertForTokenClassification,
MaskedBertForMultipleChoice,
)