Python PrepareBertMasks Beispiele

Beispiel #1

Datei: modeling_roberta.py Projekt: Tencent/TurboTransformers

 def __init__(self, embeddings: TorchRobertaEmbeddings,
              encoder: BertEncoder, pooler: BertPooler,
              config: RobertaConfig):
     self.config = config
     self.embeddings = embeddings
     self.encoder = encoder
     self.pooler = pooler
     self.prepare = cxx.PrepareBertMasks()

Beispiel #2

Datei: modeling_bert.py Projekt: yayuanzi8/TurboTransformers

 def __init__(self, embeddings: BertEmbeddings, encoder: BertEncoder):
     self.embeddings = embeddings
     self.encoder = encoder
     self.prepare = cxx.PrepareBertMasks()

Beispiel #3

 def __init__(self, embeddings: TorchBertEmbeddings,
              encoder: BertEncoderSmartBatch):
     self.embeddings = embeddings  #torch
     self.encoder = encoder
     self.prepare = cxx.PrepareBertMasks()