modelzoo.transformers.pytorch.bert.bert_model#

Classes

BertModel

The model behaves as a bidirectional encoder (with only self-attention), following the architecture described in Attention is all you need by Ashish Vaswani, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N.

BertPooler