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()
def __init__(self, embeddings: BertEmbeddings, encoder: BertEncoder): self.embeddings = embeddings self.encoder = encoder self.prepare = cxx.PrepareBertMasks()
def __init__(self, embeddings: TorchBertEmbeddings, encoder: BertEncoderSmartBatch): self.embeddings = embeddings #torch self.encoder = encoder self.prepare = cxx.PrepareBertMasks()