Module: Transformers::Bert

Defined in:

lib/transformers/models/bert/modeling_bert.rb,
lib/transformers/models/bert/tokenization_bert.rb,
lib/transformers/models/bert/configuration_bert.rb,
lib/transformers/models/bert/tokenization_bert_fast.rb

Defined Under Namespace

Classes: BertAttention, BertConfig, BertEmbeddings, BertEncoder, BertForMaskedLM, BertForSequenceClassification, BertForTokenClassification, BertIntermediate, BertLMPredictionHead, BertLayer, BertModel, BertOnlyMLMHead, BertOutput, BertPooler, BertPreTrainedModel, BertPredictionHeadTransform, BertSelfAttention, BertSelfOutput, BertTokenizer, BertTokenizerFast

Constant Summary

BERT_SELF_ATTENTION_CLASSES =

{
  "eager" => BertSelfAttention
}